hdu 5073 Galaxy 2014 Asia AnShan Regional Contest

最新推荐文章于 2018-12-03 10:00:00 发布

I-Hsin

最新推荐文章于 2018-12-03 10:00:00 发布

阅读量528

点赞数

文章标签： acm 数学

本文链接：https://blog.csdn.net/ihsin/article/details/46873481

版权

这一题我开始YY的算法是贪心==但后来感觉不对。

因为将某个质点移到质心处产生的动量是0，所以最后的质点系一定是连在一起的中间一段，因为这样各个质点距离质心越近。

质心的公式是

$x_m=\frac{\sum_{i=1}^nw_ix_i}{\sum_{i=1}^nw_i}=\frac{\sum_{i=1}^nx_i}{n}$

显然移动到质心的质点越多最后的动量也就越小，所以最后剩余 $n-k$ 个质点，对应的质心公式为 $x_m=\frac{\sum_{i=l}^r x_i}{n-k}$

对应的动量为

$I=\sum_{i=l}^r (x_i-x_m)^2\\ =\sum_{i=l}^r (x_i^2-2x_ix_m+x_m^2)\\ =\sum_{i=l}^r x_i^2-2x_m\sum_{i=l}^r x_i+(n-k)x_m^2\\ =\sum_{i=l}^r x_i^2-2\frac{(\sum_{i=l}^r x_i)^2}{n-k}+\frac{(\sum_{i=l}^r x_i)^2}{n-k}\\ =\sum_{i=l}^r x_i^2-\frac{(\sum_{i=l}^r x_i)^2}{n-k}\\ =sum_2-\frac{sum_1^2}{n-k}$

所以只要从左到右遍历所有的[l,r]区间找到最小动量的区间即可。当然首先需要将坐标排序，我开始给忘了><

注意int会溢出，要用long long，还要注意精度。

另外， $n=k$ 的情况要特判，因为遍历的时候是从 $l=1$ 开始，对于 $n=k$ 的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公式编辑器用来插入公式好爽啊~\(≧▽≦)/~