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

문제

---

https://school.programmers.co.kr/learn/courses/30/lessons/12927

풀이

---

우선순위 큐를 사용하는 문제로, 숫자가 클 수록 제곱 값이 커지므로, 가장 큰 수를 1 빼주는 것을 n번 반복한다.

#include <string>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;

long long solution(int n, vector<int> works)
{
    long long answer = 0;

    priority_queue<int> pq;

    for (int i = 0; i < works.size(); i++)
    {
        pq.push(works[i]);
    }

    while (n != 0)
    {
        if (pq.top() == 0)
        {
            return 0;
        }

        int max = pq.top() - 1;
        pq.pop();
        pq.push(max);
        n--;
    }

    while (!pq.empty())
    {
        answer += (pq.top() * pq.top());
        pq.pop();
    }

    //580
    return answer;
}