응급실

설명

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

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

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

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

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

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

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

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

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

 

입력

첫 줄에 자연수 N(5<=N<=100)과 M(0<=M<N) 주어집니다.

두 번째 줄에 접수한 순서대로 환자의 위험도(50<=위험도<=100)가 주어집니다.

위험도는 값이 높을 수록 더 위험하다는 뜻입니다. 같은 값의 위험도가 존재할 수 있습니다.

 

출력

M번째 환자는 몇 번째로 진료받는지 출력하세요.

예시 입력 1 

5 2
60 50 70 80 90

 

예시 출력 1

3

 

예시 입력 2 

6 3
70 60 90 60 60 60

 

예시 출력 2

4

 

소스 코드 1

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
// 응급실
class Person{
    int id;
    int priority;
    public Person(int id, int priority) {
        this.id = id;
        this.priority = priority;
    }
}

public class Main {
    public static int solution(int N, int M, int[] dangers) {
        int ans = 0;
        Queue<Person> queue = new LinkedList<>();
        for (int i = 0; i < N; i++) {
            queue.offer(new Person(i, dangers[i])); // 번호, 위험도로 객체 생성해서 큐에 삽입
        }

        while (!queue.isEmpty()) { //queue가 모두 빌 때까지 반복
            Person tmp = queue.poll(); //빼려고 하는 원소를 tmp로 선언
            int max = tmp.priority;
            for (Person x : queue) {
                if (max < x.priority) { //tmp 보다 큰 원소가 있는지 확인
                    queue.offer(tmp);
                    tmp = null;
                    break;
                }
            }

            if (tmp != null) {
                ans++; // 대기목록 환자 한 명 빠질 때 마다 1씩 ans에 카운트
                if (tmp.id == M) { //id=M 이 되는 조건을 만족해야 결과 리턴
                    return ans;
                }
            }
        }

        return ans;
    }


    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int M = sc.nextInt();
        int[] dangers = new int[N];
        for (int i = 0; i < N; i++) {
            dangers[i] = sc.nextInt();
        }

        System.out.println(solution(N,M,dangers));
    }
}