2019牛客多校第一场C

2019牛客多校第一场C题

题目大意：

给定$n$维空间中的一个点$A(\frac{a_1}{m},\frac{a_2}{m},\dots ,\frac{a_n}{m})$,$a_i$和$m$都是整数。现在，我们需要确定一个点$P=(p_1,p_2,\dots ,p_n)$满足：
1、$p_1,p_2,\dots ,p_n$都是实数
2、$p_i\ge 0$
3、$p_1+p_2+\dots +p_n=1$
4、使得$\sum _{i=1}^n(\frac{a_i}{m}-p_i{)}^2$最小

思路

$m$是定值，可以先不管，记$P_i=m\ast p_i$。现在要求的是$\sum _{i=1}^n(a_i-P_i{)}^2$的最小值，可以观察到$(a_i-P_i)$只会随着$P_i$的增大而减小，因为$(P_i\ge 0)$。先将$a_i$从大到小排个序，对于每个i，如果可以使$(a_j-P_j)==a_{i+1}，j\in [1,i]$,且$m-\sum _{j=1}^i{P}_j\ge 0$,这样可以将大的部分的值降下来，这样可以找到一个点$id$，其之前的$a_i$都被降到了值$a_{i}d$,之后的$P_i=0$,即后面的$a_i$值不变。此时，还有剩下的一些部分，可以将其平均分配到前$id$项，记这个部分是$res$。那么最终的结果为$\frac{id\ast (a_{id}-\frac{res}{id}{)}^2+\sum _{i=id+1}^n{a}_i^2}{m^2}$。这个做法最重要的思想就是$P$只会使得$a$减小，所以优先使最大值降低是最优的（不管这个最大值此时是正值还是负值）。

#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 1e4+10;
int n,m;
int a[maxn];
ll gcd(ll a,ll b){
	return b==0?a:gcd(b,a%b);
}
int main(){
	while(scanf("%d%d",&n,&m)!=EOF){
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			a[i]=-a[i];
		}
		sort(a+1,a+1+n);
		for(int i=1;i<=n;i++)a[i]=-a[i];
		int res=m,id=1;
		ll x,y;
		for(id=1;id<n;id++){
			if(id*abs(a[id+1]-a[id])<=res){
				res-=id*abs(a[id+1]-a[id]);
			}
			else {
				break;
			}
		}
		ll s=0;
		for(int i=id+1;i<=n;i++)s+=a[i]*a[i];
		x=(a[id]*id-res)*(a[id]*id-res)+id*s;
		//printf("%d\n",id);
		y=m*m*id;
		//printf("%lld/%lld\n",x,y);
		ll g=gcd(x,y);
		x=x/g;y=y/g;
		if(y==1)printf("%lld\n",x);
		else printf("%lld/%lld\n",x,y);
	}
	return 0;
}