这一题我开始YY的算法是贪心==但后来感觉不对。
因为将某个质点移到质心处产生的动量是0,所以最后的质点系一定是连在一起的中间一段,因为这样各个质点距离质心越近。
质心的公式是
显然移动到质心的质点越多最后的动量也就越小,所以最后剩余个质点,对应的质心公式为
对应的动量为
所以只要从左到右遍历所有的[l,r]区间找到最小动量的区间即可。当然首先需要将坐标排序,我开始给忘了><
注意int会溢出,要用long long,还要注意精度。
另外,的情况要特判,因为遍历的时候是从开始,对于的case不一样。
#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<stdlib.h>
#include<vector>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<ctype.h>
#include<map>
#include<time.h>
#include<bitset>
#include<set>
#include<list>
using namespace std;
//hdu 5073
const int maxn=50010;
const double esp=1e-10;
int T;
int n;
int k;
long long x[maxn];
long long sum1;
long long sum2;
long long ans;
int main()
{
freopen("input.txt","r",stdin);
//freopen("out1.txt","w",stdout);
scanf("%d",&T);
for(int ca=1;ca<=T;ca++)
{
memset(x,0,sizeof(x));
sum1=0;
sum2=0;
ans=0;
scanf("%d %d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%I64d",&x[i]);
}
if(n==k)//||k==(n-1)
{
printf("0\n");
continue;
}
sort(x+1,x+n+1);
for(int i=1;i<=(n-k);i++)
{
sum1+=x[i];
sum2+=x[i]*x[i];
}
ans=(n-k)*sum2-sum1*sum1;
int left=1;
int right=n-k+1;
for(int i=n-k+1;i<=n;i++)
{
long long tmp=(n-k)*(sum2-x[left]*x[left]+x[right]*x[right])-(sum1-x[left]+x[right])*(sum1-x[left]+x[right]);
if(ans>tmp)
{
ans=tmp;
}
sum2=sum2-x[left]*x[left]+x[right]*x[right];
sum1=sum1-x[left]+x[right];
left++;
right++;
}
printf("%.10lf\n",1.0*ans/(n-k));
//cout<<1.0*ans/(n-k)<<endl;
}
return 0;
}
PS:我发现在线LaTex公式编辑器用来插入公式好爽啊~\(≧▽≦)/~