题目
分析
将每个物品都放到大根堆里,但每个物品有两个信息:长度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;
}