[JAVA] 알고리즘 : Queue- 응급실

import java.util.*;

class Person{
    int id; // 순서
    int priority; // 위험도
    public Person(int id, int priority){
        this.id=id;
        this.priority=priority;
    }
}
public 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));
    }
}

 

 

 

 

환자들의 위험도를 큐에 세팅한다.

큐의 맨앞의 환자를 꺼내고 for문을 돌며 다른 환자들의 위험도랑 비교한다.

맨 앞의 환자보다 위험도가 큰 환자가 있으면 맨 앞의 환자를 큐의 뒤에 추가한다.

맨 앞의 환자의 위험도가 가장 크다면 answer를 증가시키고 answer가 id(기존 순서)가 구하려는 환자의 번호와 같다면 리턴시킨다.