题意
数轴上有m个生产车间可以生产零件。一共有n种零件,编号为1~n。第i个车间的坐标为xi,生产第pi种零件(1<=pi<=n)。你需要在数轴上的某个位置修建一个组装车间,把这些零件组装起来。为了节约运输成本,你需要最小化cost(1)+cost(2)+…+cost(n),其中cost(x)表示生产第x种零件的车间中,到组装车间距离的平方的最小值。
n<=100000
分析
满脑子的骚操作结果一点用都没有。。。
挺巧妙的一道题。
显然是要最小化
∑i=1n(x−xi)2
拆开得到
n∗x2+∑i=1nx2i−2x∗∑i=1nxi
显然当x=∑i=1nxin的时候取最小
且最小值等于
∑i=1nx2i−∑i=1nxin
那么我们可以先把每种零件的最左边的点设为最优点,然后每次找到某种零件当前点与其下一个点的中点的最小值,然后替换一下,最后取个最优值即可。
代码
#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;
}