C++ Using Max Heap 15 Lines of Code


#1
int Solution::solve(vector<int> &A, int B) {
    priority_queue<int> p;
    for(auto i:A)
        p.push(i);
    int ans=0,temp;
    while(B--)
    {
        temp=p.top();
        ans+=temp;
        p.pop();
        p.push(temp-1);
    }
    return ans;
}