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

안녕하세요  !

PrioryQueue 오름차순 정렬 하셔서 풀이 하셨는데, 

오름차순이 아닌 내림차순으로 해서 PQ에 넣는 방법에 대해서는 어떻게 생각하시나요?

  public int solve(int[] nums , int k) {
    int answer = 0;
    // priority queue 내림차순 정렬
    PriorityQueue<Integer> pq = new PriorityQueue(Collections.reverseOrder());

    // Priority Queue에 배열 원소들을 모두 넣는다.
    for (int i = 0; i < nums.length; ++i) {
      pq.offer(nums[i]);
    }

    // k개 꺼내서, 마지막으로 꺼낸 원소를 반환한다
    while(k > 0) {
      answer = pq.poll();
      --k;
    }

    return answer;
  }