프로그래머스 - 야근 지수 (구현?)

쉬운데 5점이나 줫다

풀면서 문제가 이렇게 쉽다고? 생각했는데 진짜 쉬웟다

이게 기둥과보랑 같이 있는 문제라는 것이 믿겨지지 않는다.

처음에는 Solution으로 풀었는데 (반복문 쓰면 통과 못할까봐)

그냥 Solution1처럼 반복문 쓰고 풀어도 풀린다

    public long solution1(int n, int[] works) {
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
 
        long answer = 0;
        long sum = 0;
 
        for (int i = 0; i < works.length; i++) {
            pq.add(works[i]);
            sum += works[i];
        }
        if (sum - n <= 0)
            return 0;
 
        while (true) {
            int now = pq.poll();
            pq.add(now - 1);
            if (n-- == 1)
                break;
        }
 
        while (!pq.isEmpty()) {
            int now = pq.poll();
            answer += (now * now);
            // System.out.println(pq.poll());
        }
 
        return answer;
    }
    
    // fail
    public long solution(int n, int[] works) {
        long answer = 0;
 
        // Arrays.sort(works);
 
        long sum = 0;
        int leng = works.length;
 
        for (int i = 0; i < works.length; i++) {
            sum += works[i];
            // System.out.println(works[i]);
        }
        long minSum = sum - n; // 일 총합에서 야근뺀것.. 이걸로 야근지수 만듬..
        long aver = minSum / leng;
 
        if (minSum <= 0)
            return 0;
 
        // System.out.println();
        // System.out.println(minSum);
        // System.out.println(aver);
        // System.out.println(aver*leng);
 
        long plusOneCount = minSum - aver * leng;
 
        answer = (aver * aver) * (leng - plusOneCount) + (aver + 1) * (aver + 1) * (plusOneCount);
        // System.out.println(leng-plusOneCount);
        // System.out.println(plusOneCount);
 
        return answer;
    }