题意:
给你n个数,要你把n个数切成k个。求切后最小平方和。
题解
一开始就想着把每个数放进优先队列里,然后把最大的平分,再放回去。但这样是错的。如1 3 100 :33 33 34会更好。
用优先队列维护结构体{a,b,c}。b存的是读入的数。a表示给这个数切(c刀-c+1刀)这个数的平方差值,c代表现在切的刀数。并对a进行大到小排序。
反正就是一刀一刀的切,取差值最大的那个。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
struct node
{
ll a,b,c;
friend bool operator<(node x,node y)
{
return x.a<y.a;
}
};
priority_queue<node>q;
ll cut(int a,int x)//把a切成x刀,返回平方差值
{
ll m=a/x,t=a%x;//t代表有t个数是m+1。。。。(x-t)个数是m
return t*(m+1)*(m+1)+(x-t)*m*m;
}
int main()
{
int n,k;
cin>>n>>k;
ll ans=0;
for(int i=1;i<=n;i++)
{
ll x;
cin>>x;
ans+=x*x;
q.push({cut(x,1)-cut(x,2),x,2});
}
for(int i=n;i<k;i++)
{
auto x=q.top();
q.pop();
ll a=x.a,b=x.b,c=x.c;
ans-=a;
q.push({cut(b,c)-cut(b,c+1),b,c+1});
}
cout<<ans;
return 0;
}