인프런 커뮤니티 질문&답변

- 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!
- 먼저 유사한 질문이 있었는지 검색해보세요.
- 서로 예의를 지키며 존중하는 문화를 만들어가요.
- 잠깐! 인프런 서비스 운영 관련 문의는 1:1 문의하기를 이용해주세요.

위험도가 높은 환자 순서로 우선순위 큐를 만들었습니다. 정답으로 찾아야 하는 환자의 위치의 번호를 이동해가면서 현재 치료 받는 환자가 찾아야 하는 환자인지를 체크 했고요. 우선순위 큐에서 poll한 값을 통해서 현재 대기하고 있는 환자가 현재 치료를 받아도 되는지(우선순위가 가장 높은지)를 체크 했습니다.

public int solution(int[] input1, int m) {
        Queue<Integer> q = new LinkedList<>();
        Queue<Integer> pq = new PriorityQueue<>(Comparator.reverseOrder());
        int seq = m;

        for (int i : input1) {
            q.offer(i);
            pq.offer(i);
        }

        int cnt = 0;

        while (true) {
            Integer poll = q.poll();
            seq--;

            if (Objects.equals(pq.peek(), poll)) {
                cnt++;

                if (seq == -1) {
                    break;
                }

                pq.poll();
            } else {
                q.offer(poll);

                if (seq == -1) {
                    seq = q.size() - 1;
                }
            }
        }

        return cnt;
    }