설명

메디컬 병원 응급실에는 의사가 한 명밖에 없습니다.

응급실은 환자가 도착한 순서대로 진료를 합니다. 하지만 위험도가 높은 환자는 빨리 응급조치를 의사가 해야 합니다.

이런 문제를 보완하기 위해 응급실은 다음과 같은 방법으로 환자의 진료순서를 정합니다.

• 환자가 접수한 순서대로의 목록에서 제일 앞에 있는 환자목록을 꺼냅니다.

• 나머지 대기 목록에서 꺼낸 환자 보다 위험도가 높은 환자가 존재하면 대기목록 제일 뒤로 다시 넣습니다. 그렇지 않으면 진료를 받습니다.

즉 대기목록에 자기 보다 위험도가 높은 환자가 없을 때 자신이 진료를 받는 구조입니다.

현재 N명의 환자가 대기목록에 있습니다.

N명의 대기목록 순서의 환자 위험도가 주어지면, 대기목록상의 M번째 환자는 몇 번째로 진료를 받는지 출력하는 프로그램을 작성하세요.

대기목록상의 M번째는 대기목록의 제일 처음 환자를 0번째로 간주하여 표현한 것입니다.

코드

public class EmergencyRoom {

    public static void main(String[] args){
        Scanner in=new Scanner(System.in);
        int input1 = in.nextInt();
        int input2 = in.nextInt();
        int[] arr = new int[input1];
        for(int i=0; i<input1; i++){
            arr[i] = in.nextInt();
        }

        int solution = solution(arr, input2);
        System.out.println(solution);
    }

    public static int solution(int[] arr, int target){
        int answer = 1;
        Queue<Person> que = new LinkedList<>();
        for(int i =0; i<arr.length; i++){
            que.offer(new Person(i, arr[i]));
        }

        //진료 시작
        while(!que.isEmpty()){
            Person tmp = que.poll();
            for(Person p : que){    //나머지 que와 비교하기
                if(p.priority > tmp.priority){
                    que.offer(tmp);
                    tmp = null;
                    break;
                }
            }
            //진료를 보는 환자
            if(tmp != null){
                if(tmp.id == target) return answer;
                else answer++;
            }
        }

        return answer;
    }
    static class Person{
        int id;
        int priority;

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

Queue를 아직 제대로 이해하지 못했거나 문제를 풀어내는 방법을 아직 잘 찾아내지 못하는 것 같다. que에 넣어놓고 poll한 순간 다른 que와 비교할 수 있다는 생각을 못하고 어떻게 나머지 값들과 비교를 해야할까에서 막혀서 문제를 풀고 있지 못했는데. 그냥 que를 poll하고 그 값을 갖고 나머지 que를 모두 돌려서 비교하면 되었던 것이다... 이제 이 개념을 이해했으니 다른 que문제에서는 활용할 수 있기를 바란다.

Queue에서 poll()하면 나머지 값들과 비교해볼 수 있다.