[Wikioi 1135][NOIP 2011提高组]选择客栈

题目描述 Description

丽江河边有 n 家很有特色的客栈，客栈按照其位置顺序从1 到n 编号。每家客栈都按照某一种色调进行装饰（总共k 种，用整数0 ~ k-1 表示），且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。
两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈），且咖啡店的最低消费不超过p。
他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过p元的咖啡店小聚。

输入描述 Input Description

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

输出描述 Output Description

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

样例输入 Sample Input

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

样例输出 Sample Output

3

数据范围及提示 Data Size & Hint

【输入输出样例说明】

客栈编号	①	②	③	④	⑤
色调	0	1	0	1	1
最低消费	5	3	2	4	5

2 人要住同样色调的客栈，所有可选的住宿方案包括：住客栈①③，②④，②⑤，④⑤，
但是若选择住 4、5 号客栈的话，4、5 号客栈之间的咖啡店的最低消费是 4，而两人能承受
的最低消费是 3 元，所以不满足要求。因此只有前 3 种方案可选。

 

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

数据统计 Statistics

总算把搁置一个多月的这道题AC掉了，题解看代码注释，这个题貌似解法很多，反正范围太大了，O(n^2)的算法绝对过不了，有童鞋用暴搜+记忆化能拿九十分，不过此题最好的方法还是用DP，这里我用的是O(2n-1)的DP，按照终点(第二个客栈的位置)进行DP，在DP过程中要注意下如何获得当前客栈前，与当前客栈色调相同的客栈数等等，最好在每次循环时用数组保存这些信息，避免决策前还跑一层循环，导致复杂度增加一维

#include <stdio.h>

#define MAXN 200500

int f[MAXN],r[MAXN],color[MAXN],maxc[MAXN],c1[MAXN],c2[MAXN];
/*
设f[i]为第1~i的客栈中编号最大的最低消费不大于p的编号,r[i]为第1~i-1的客栈中
编号最大的色调与i客栈相同的编号,c1[i]=1~i-1号旅馆中与第i号旅馆色调相同的
客栈数目,c2[i]=1~i-1号客栈中与第i号客栈色调相同,且到第i号旅馆路上存在最低消费不大于p的客栈的客栈数
目,输出答案为Σc2[i],i∈(1,n]
(i)若f[i]<r[i],则1~r[i]号客栈中编号最大的最低消费小于p的客栈编号与1~i的相同,
   即f[i]=f[r[i]],c2[i]=c2[r[i]]
(ii)若f[i]>=r[i],则1~r[i]路上,所有与i号客栈颜色相同的客栈到i号客栈路上必
   然存在一个旅馆的最低消费不大于p,
   c2[i]=c1[i]

  color[i]=之前所有客栈中，色调为i的客栈个数,maxc[i]=之前所有客栈中，色调为i的最大编号
*/

int main()
{
	int i,j,n,k,p,a,b,ans=0;
	scanf("%d%d%d",&n,&k,&p);
	for(i=1;i<=n;i++)
	{
		scanf("%d%d",&a,&b);
		r[i]=maxc[a];
		if(b<=p) f[i]=i;
		else f[i]=f[i-1];
		c1[i]=color[a];
		if(f[i]<r[i]) c2[i]=c2[r[i]];
		else c2[i]=c1[i];
		maxc[a]=i;
		color[a]++;
	}
	for(i=2;i<=n;i++) ans+=c2[i];
	printf("%d\n",ans);
	return 0;
}