글 읽기 - 1966번, 큐 관련된 문제 질문입니다 ..ㅠㅠ

jinkor 7년 전 0

제생각엔 코드는 정확하게 쓴것같은데 시간초과라고 나오네요.

문제가 원하는 해결방향(알고리즘)이 다른건가요? 있다면 무엇인가요?

import java.util.Comparator;
import java.util.LinkedList;
import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Scanner;

class Temp{
	int id;
	int value;
	Temp(int id, int value){
		this.id = id;
		this.value = value;
	}
	public String toString() {
		return "("+id+","+value+")";
	};
}

public class Main {
	public static Scanner scanner = new Scanner(System.in);
	public static PriorityQueue<Temp> pQueue = new PriorityQueue<>(new Comparator<Temp>(){
		@Override
		public int compare(Temp o1, Temp o2) {
			if(o1.value > o2.value){
				return -1;
			} else if( o1.value < o2.value){
				return 1;
			} else{
				return 0;
			}
		}
	});
	
	public static boolean isMax(Queue<Temp> queue,Temp input){
		/*
		for(Temp item : queue){
			if(input.value <= item.value){
				return false;
			}
		}
		return true;
		*/
		if(pQueue.peek().value <= input.value){
			return true;
		} else{
			return false;
		}
	}
	
	public static void main(String[] args) {
		int testCase;
		testCase = scanner.nextInt();
		TestUnit:
		for(int i = 0 ; i < testCase ; i++){
			Queue<Temp> queue = new LinkedList<>();
			int N; // the number of documents
			int id; // target item
			N =scanner.nextInt();
			id = scanner.nextInt();
			for(int j = 0 ; j < N ;j++){
				Temp temp = new Temp(j,scanner.nextInt());
				//System.out.println(temp);
				queue.add(temp);
				pQueue.add(temp);
			}
			
			int count = 0; // count 번째 pop에서 target이 발견되었다!
			while(true){
				Temp item = queue.poll();
				if(isMax(queue,item)){
					pQueue.poll();
					count++;
					//System.out.println("count: "+count+" item: "+item);
					if(item.id == id){
						System.out.println(count);
						continue TestUnit;
					}
				} else{
					queue.add(item);
				}
			}
		}
	}
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

1966번, 큐 관련된 문제 질문입니다 ..ㅠㅠ

jinkor 7년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말