Codeforces Raif Round 1 (Div. 1 + Div. 2) E. Carrots for Rabbits（优先队列+贪心）

原题链接
题意
有N个萝卜，K只兔子，要求把N个萝卜分成K份，一个兔子吃一个长为X的萝卜的时间为X*X，求怎么分能使最后的时间最短。
思路
刚看到这个题的时候，刚开始的思路是，把所有的萝卜都存在一个大根堆里面，然后每次取堆头元素分为两份，直到最后大根堆中的数量刚好等于K，但是交上去以后发现WA了，又想了一下，确实不能这样，因为你不确定每一个萝卜分成几段，比如10这根萝卜，第一次分成5，5，第二次再分是5，3，2，但是这样分并不是最小的，因为3，3，4这样分是最小的。
最后看了大佬的思路以后才想起来，先把现有萝卜吃完所需要的时间是

ans+=(a[i]*a[i]+(a[i+1]*a[i+1]+.....

然后用一个大根堆存一个萝卜划分为1，到划分为2的差值，每一个都存进来，那一个差值最大，就先分那一个，然后再把这个萝卜从2划分为3的差值存进优先队列，依次找k-n个值。
代码

#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
const int N=1e5+10;
typedef long long ll;
ll a[N];
struct Node
{
    ll val;  // 这一次与下一次的差值
    ll now;
    ll nex;
    ll id;
    ll num;
    bool operator < (const Node &r) const
    {
        return val<r.val;
    }
};
ll f(int id,int num)
{
    if(num>=a[id]) return 1e18;
    num++;
    ll em=a[id]/num;
    ll ym=a[id]%num;
    ll ans=0;
    for(int i=1 ; i<=num-ym ; i++) ans+=(em*em);
    for(int i=1 ; i<=ym ; i++) ans+=((em+1)*(em+1));
    return ans;
}
priority_queue<Node> q ;
int main()
{
    int n,k;
    cin>>n>>k;
    ll ans=0;
    for(int i=1 ; i<=n ; i++)
    {
        cin>>a[i];
        ll now=a[i]*a[i];
        ll nex=f(i,1);
        q.push(Node{now-nex,now,nex,i,1});
        ans+=now;
    }
    int num=k-n;
    while(num--)
    {
        Node tp=q.top();
        q.pop();
        ans-=tp.val;
        ll nex=f(tp.id,tp.num+1);
        q.push(Node{tp.nex-nex,tp.nex,nex,tp.id,tp.num+1});
    }
    cout<<ans<<endl;
    return 0;
}