noip2011提高组——选择客栈

对于一个符合要求的点  在其两侧找到两个相同颜色客栈
我的第一反应  分治。。   （虽然可以dp  可以线段树。。）
分治一类最好使用递归实现
比较方便  递归写的时候只管当前的状态（普遍性）  再设置边界  就好了   

readdata时记录下符合要求  即价格合适的点的位置 以及数量
int rec[200000+10];//价格合适的点的位置  由小到大  
int cnt;//记录个数  
在预处理前缀和 

int now[200000+10][51];//截至第i格位置   为j颜色的有多少个客栈 

友情提示  极限情况会爆int   切勿作死
and windows系统下long long 为I64d    Linux系统下为lld

分治部分伪代码
 void divide(左右标记)
{
    if边界
    {
        特判
        返回        
    }
    递归左
    递归右
    for颜色
    ans+=左区间同颜色数*右区间同颜色数；
    ans--//防止 分界点本身的自己选择自己
} 

输出ans就对了O(∩_∩)O~

/*类nlogn算法   找到一个价格合适的点  分治左右  */
#include<iostream> 
#include<cstdio>
using namespace std;
int n,k,p;
int now[200000+10][51];//截至第i格位置   为j颜色的有多少个客栈 
int rec[200000+10];//价格合适的点的位置  由小到大
int cnt;//记录个数
struct edge
{
	int col;
	int pri;
}node[200000+10];
long long ans;
void divide(int lp,int rp,int l,int r) //一段区间里开头的分界点 lp  结尾分界点rp 
{
	int midp=(lp+rp)/2;
	if(lp>rp)
	return;
	if(lp==rp)
	{
		if(l==r)
		return;
		for(int i=0;i<=k;i++)
		ans+=(now[rec[midp]][i]-now[l-1][i])*(now[r][i]-now[rec[midp]-1][i]);
		ans--;
		return;
	}
	divide(lp,midp-1,l,rec[midp]-1);
	divide(midp+1,rp,rec[midp]+1,r);
	for(int i=0;i<=k;i++)
	ans+=(now[rec[midp]][i]-now[l-1][i])*(now[r][i]-now[rec[midp]-1][i]);
	ans--;	
}
void readdata()
{
	scanf("%d%d%d",&n,&k,&p);
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&node[i].col,&node[i].pri);
		if(node[i].pri<=p)
		rec[++cnt]=i;
		for(int j=0;j<=k;j++)
		{
			now[i][j]=now[i-1][j];
			if(node[i].col==j)
			now[i][j]++;
		}
	}
}
int main()
{
	freopen("hotel.in","r",stdin);
	freopen("hotel.out","w",stdout);
	readdata();
	divide(1,cnt,1,n);
	printf("%I64d\n",ans);
}