할 수 있다는 신념, 하면 된다는 생각

8. 응급실

5 2
60 50 70 80 90

3

6 3
70 60 90 60 60 60

4

접근방법

Queue를 활용하여 풀 건데, 이전과는 다르게 큐에 값을 하나만 넣는 것이 아니라 두 개를 넣어야 한다는 문제가 있다.

보통, 큐나 스택을 설명하면 아래의 그림처럼 하나의 필드만을 얘기하는데, 두 필드를 활용하려면 어떻게 해야 할까?

바로, 객체를 생성하여 큐에 객체의 주소를 집어넣는 것이다.

이렇게 하면, 주어진 m과 id가 같을 때의 값을 출력하면 된다.

import java.util.*;

class Person{
	int id;
	int priority;
	public Person(int id, int priority) {
		this.id = id;
		this.priority = priority;
	}
}

class Main{
	
	public int solution(int n, int m, int[] arr) {
		int answer = 0;
		Queue<Person> Q = new LinkedList<>();
		for(int i = 0; i < n; i++) {
			Q.offer(new Person(i, arr[i]));
		}
		
		while(!Q.isEmpty()) {
			Person tmp = Q.poll();
			for(Person x : Q) {
				if(x.priority > tmp.priority) {
					Q.offer(tmp);
					tmp = null;
					break;
				}
			}
			if(tmp!=null) {
				answer++;
				if(tmp.id == m)	return answer;
			}
			
		}
		
		return answer;
	}
	
	public static void main(String args[]) {
		
		Main T = new Main();
		
		Scanner sc = new Scanner(System.in);

		int n = sc.nextInt();
		int m = sc.nextInt();
		int[] arr = new int[n];
		for(int i = 0; i < n; i++)	arr[i] = sc.nextInt();
		System.out.println(T.solution(n, m, arr));
		
	}		
}