E. Carrots for Rabbits（一种神奇的数列维护技巧）

题目

E. Carrots for Rabbits

分析

将每个物品都放到大根堆里，但每个物品有两个信息：长度len、当前被切了几段part。根据这个物品当前状态的贡献作为在堆里位置的依据。贡献由cal函数算出。

注意：一个物品切成x段时，应保证这x段的最长段和最短段的差值尽可能的小。而事实上，可以保证最长段和最短段的差值为1或0。

代码

#include<bits/stdc++.h>
#define ll long long
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define no cout<<"No\n"
#define yes cout<<"Yes\n"
#define pii pair<int, int>
#define pll pair<ll, ll>
#define sol void solve
using namespace std;
int n,k; 
struct node{
    ll len;
    ll part;
};
ll cal(ll le, ll par){
    ll tem=le/par;
    ll mod=le%par;
    ll ans=(par-mod)*tem*tem;
    ans+=mod*(tem+1)*(tem+1);
    return ans;
}
bool operator< (const node &a,const node &b) {
    return cal(a.len, a.part) - cal(a.len, a.part + 1) < cal(b.len, b.part) - cal(b.len, b.part + 1);
}
sol(){
    priority_queue<node> q;
    cin>>n>>k;
    ll x, ans=0;
    rep(i,1,n){
        cin>>x;
        ans+=x*x;
        q.push({x,1});
    }
    node ttt;
    rep(i,1,k-n){
        ttt=q.top();
        q.pop();
        ans -= cal(ttt.len, ttt.part) - cal(ttt.len, ttt.part+1);
        q.push({ttt.len, ttt.part+1});
    }
    cout<<ans;
}
int main(){
    ios_base::sync_with_stdio(false); 
    cin.tie(0); 
    cout.tie(0);
    int t=1;
//    cin>>t;
    while(t--){
        solve();
    }
    return 0;
}