[c++] 야근 지수

#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#define ll long long 
using namespace std;
 
long long solution(int n, vector<int> works) {
    long long answer = 0;
    sort(works.begin(),works.end());
    int height=works.back();
    reverse(works.begin(),works.end());
    
    
    while(height && n){
        for(int i=0;i<works.size();i++){
            if(works[i]==height &&n>0){
                works[i]--;
                n--;
            }
            else if(works[i]<height)
                break;
        }
        height--;
    }
    for(int i=0;i<works.size();i++){
        answer+=(works[i]*works[i]);
    }
    return answer;
}