bzoj 2666: [cqoi2012]组装 贪心

题意

数轴上有m个生产车间可以生产零件。一共有n种零件，编号为1~n。第i个车间的坐标为xi，生产第pi种零件（1<=pi<=n）。你需要在数轴上的某个位置修建一个组装车间，把这些零件组装起来。为了节约运输成本，你需要最小化cost(1)+cost(2)+…+cost(n)，其中cost(x)表示生产第x种零件的车间中，到组装车间距离的平方的最小值。
n<=100000

分析

满脑子的骚操作结果一点用都没有。。。

挺巧妙的一道题。
显然是要最小化

$$\sum_{i=1}^n(x-x_i)^2$$
拆开得到 $$n*x^2+\sum_{i=1}^nx_i^2-2x*\sum_{i=1}^nx_i$$
$$显然当x=\frac{\sum\limits_{i=1}^nx_i}{n}的时候取最小$$
且最小值等于 $$\sum_{i=1}^nx_i^2-\frac{\sum\limits_{i=1}^nx_i}{n}$$
那么我们可以先把每种零件的最左边的点设为最优点，然后每次找到某种零件当前点与其下一个点的中点的最小值，然后替换一下，最后取个最优值即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=100005;

int n,m,now[N],a1;
bool vis[N];
struct data{int x,y,val;}a[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

bool cmp(data a,data b)
{
    return a.val<b.val;
}

int main()
{
    n=read();m=read();
    double s1=0,s2=0;
    for (int i=1;i<=m;i++)
    {
        int x=read(),p=read();
        if (vis[p]) a[++a1].x=now[p],a[a1].y=x,a[a1].val=now[p]+x;
        else vis[p]=1,s1+=(double)x*x,s2+=(double)x;
        now[p]=x;
    }
    sort(a+1,a+a1+1,cmp);
    double ans=s1-s2*s2/(double)n,pos=s2/(double)n;
    for (int i=1;i<=a1;i++)
    {
        s1-=(double)a[i].x*a[i].x-(double)a[i].y*a[i].y;
        s2-=(double)a[i].x-(double)a[i].y;
        if (s1-s2*s2/(double)n<ans) ans=s1-s2*s2/(double)n,pos=s2/(double)n;
    }
    printf("%.4lf",pos);
    return 0;
}