noip 2011 选择客栈

描述

丽江河边有n家很有特色的客栈，客栈按照其位置顺序从1到n编号。每家客栈都按照某一种色调进行装饰（总共k种，用整数0~ k-1表示），且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。

两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈），且咖啡店的最低消费不超过p。

他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过p元的咖啡店小聚。

格式

输入格式

第一行三个整数n，k，p，每两个整数之间用一个空格隔开，分别表示客栈的个数，色调的数目和能接受的最低消费的最高值；
接下来的n行，第i+1行两个整数，之间用一个空格隔开，分别表示i号客栈的装饰色调和i号客栈的咖啡店的最低消费。

输出格式

输出只有一行，一个整数，表示可选的住宿方案的总数。

样例1

样例输入1[复制]

5 2 3
0 5
1 3
0 2
1 4
1 5

样例输出1[复制]

3

限制

1s

提示

对于30%的数据，有n≤100；
对于50%的数据，有n≤1,000；
对于100%的数据，有2≤n≤200,000，0<k≤50，0≤p≤100，0≤最低消费≤100。

【题解】根据我的老习惯，我准备写一份题解。

          这道题我一开始打算找中间的然后判断两边，但发现这样还要判重，所以觉得应该是确定两个同色客栈这样只要判断之间是否存在 <=p 的就可以了。然后我发现，如果我把最低价大于p 的 当作 0 ，小于等于 p 的当作 1 ，这样，求一下前缀就可以O(1) 判断之间是否存在。然后由于n 有 200,000,O(n^2)是肯定不行的，所以我想到可不可以O(n)解决。假设当前在做第 i 间客栈，并认为i 对答案的贡献为 i 之前与 i成一对。这样，与 i 贡献有关的就是标号在i之前并且色调相同的客栈。因为sum[] 数组一定是单调不递减的，所以我设了三个量来记录，f[ color[i] ][1] 表示一定可行的数量，f[ color[i] ][2]记录可能可以，可能不可以的（需要判断过的）数量，f[ color[i] ][3]表示color[i] 这种颜色最新一次做过的前缀和。那么贡献就是判断 f[ color[i] ][3]<=p,ans+=f[color[i] ][1]+f[color[i]][2] ;else->ans+=f[color[i]][1];

然后就可以O（n）求解了。

#include<cstdio>
#include<cstdlib>

#define maxk 51
#define maxn 200010

int sum[maxn],color[maxn],value[maxn],f[maxk+1][4];

int main()
{
	int n,k,p,ans=0;sum[0]=0;
	scanf("%d%d%d",&n,&k,&p);
	for(int i=1,x;i<=n;i++)
	{
		scanf("%d%d",&color[i],&value[i]);
		sum[i]=sum[i-1]+(value[i]<=p);//可行前缀和； 
	}
	color[0]=maxk+1;//边界判断； 
	for(int i=1;i<=n;i++)
	{
			int tot=f[color[i]][1];
			if(f[color[i]][3]<sum[i]){
				tot+=f[color[i]][2];
				f[color[i]][1]+=f[color[i]][2];
				f[color[i]][2]=0;
			}
			ans+=tot;
			f[color[i]][2]++;
			f[color[i]][3]=sum[i]-(value[i]<=p);
	}
	printf("%d\n",ans);
	system("pause");
	return 0;
}