【bzoj 2666】: [cqoi2012]组装

http://www.lydsy.com/JudgeOnline/problem.php?id=2666

这道题，容易想到按x坐标排序，然后扫一遍。。。。。
问题是区间内多个颜色相同的情况，到底选哪一个成为问题


换种思路。。。
加入我们已经确定了哪些车间，可以通过预处理在O(1)内算出代价
所以考虑枚举i和last(i)平均值，那么选哪一个就不成问题了，问题在于如何枚举。。。


注意到相同颜色的分割线很重要，可以把他看为事件点。。。。。。
也就是说每经过一个事件点，更新一次答案，这样扫一遍


排序时注意神奇的下标问题！！！！！

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
#define rep(i,l,r) for(int i=(l),_=(r);i<=_;i++)
#define per(i,r,l) for(int i=(r),_=(l);i>=_;i--)
#define MS(arr,x) memset(arr,x,sizeof(arr))
#define INE(i,u) for(int i=head[u];~i;i=e[i].next)
#define LL long long
inline const int read()
{int r=0,k=1;char c=getchar();for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;
for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';return k*r;}

const int N=100010;
const int inf=0x3f3f3f3f;
int n,m;
struct data{LL x,p,lstx;data *lst,*nxt;}a[N];
int head[N];
int cnt[N];

LL F2(data O){return O.x + O.lstx;} // i 和 last(i) 的分割线的两倍 
bool cmp(data A,data B)
{
	return F2(A) < F2(B);
}

void input()
{
	n=read(); m=read();
	a[0].x=-inf;
	rep(i,1,m)
	{
		a[i].x=read();
		a[i].p=read();
		a[i].lst=&a[head[a[i].p]];
		a[i].lstx=a[i].lst->x;
		head[a[i].p]=i;
	}
	sort(&a[1],&a[m+1],cmp);
}
void solve()
{
	int clr=0;
	double ans=1e100,pos=0;
	double sa=0,s2a=0;
	rep(l,1,m)
	{
		int p=a[l].p;
		if(!cnt[p]) clr++;
		cnt[p]++;
		sa+=a[l].x;
		s2a+=a[l].x*a[l].x;
		while(cnt[p] > 1)
		{
			double xx=a[l].lstx;
			sa-=xx;
			s2a-=xx*xx;
			cnt[p]--;
		}
		if(clr == n)
		{
			double ave=sa/n;
			double res=s2a-2*sa*ave+ave*ave*n;
			if(res<ans)
			{
				ans=res;
				pos=ave;
			}
		}
	}
	printf("%.4f\n",pos);
}

int main()
{
    freopen("B.in","r",stdin); freopen("B.out","w",stdout);
    input(),solve();
    return 0;
}