[프로그래머스] 야근 지수

https://programmers.co.kr/learn/courses/30/lessons/12927?language=java 

코딩테스트 연습 - 야근 지수

2021-09-06

---

import java.util.*;
class Solution {
    public long solution(int n, int[] works) {
        long answer = 0;
        PriorityQueue<Long> pq = new PriorityQueue<>(Comparator.reverseOrder());
        for(int i = 0; i < works.length; i++) pq.add((long) works[i]);
 
        while(!pq.isEmpty()) {
            long val = pq.poll();
            if(val - 1 > 0) pq.add(--val);
            n--;
            if(n == 0) break;
        }
        
        while(!pq.isEmpty()) {
            answer += Math.pow(pq.poll(), 2);
        }
 
        return answer;
    }
}

#문제풀이 

가장 큰 값을 계속 쳐내주는게 좋다. 그래서 priorityqueue를 reverseorder로 해놓고 poll()한 값을 계속 n만큼 깎아줬다.