[NOIP2011 提高组] 选择客栈（含加强版）题解

题目（以加强版为主）

普通版题目
加强版题目

题目描述

丽江河边有 $n$ 家很有特色的客栈，客栈按照其位置顺序从 $1$ 到 $n$ 编号。

每家客栈都按照某一种色调进行装饰（总共 $k$ 种，用整数 $0\sim k-1$ 表示），且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。

两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。

晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈），且咖啡店的最低消费不超过 $p$ 。

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

输入格式

输入共 $n+1$ 行。

第一行三个整数 $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

提示

【样例解释】

$$\begin{array}{cccccc}\text{客栈编号}& \text{①}& \text{②}& \text{③}& \text{④}& \text{⑤}\\ \text{色调}& 0& 1& 0& 1& 1\\ \text{最低消费}& 5& 3& 2& 4& 5\end{array}$$

二人要住同样色调的客栈，所有可选的住宿方案包括：住客栈①③，②④，②⑤，④⑤。

但是若选择住 ④⑤ 号客栈的话，④⑤ 号客栈之间的咖啡店的最低消费是 $4$，而两人能承受的最低消费是 $3$ 元，所以不满足要求。因此只有前 $3$ 种方案可选。

【数据范围】
对于 $25\%$ 的数据，$n\le 100$；
对于 $40\%$ 的数据，$n\le 1000$；
对于 $80\%$ 的数据，$n\le 2\times 10^5$，$k\le 50$；
对于 $100\%$ 的数据，$2\le n\le 2\times 10^6$，$1\le k\le 10^4$，$0\le p\le 100$，$0\le$ 最低消费 $\le 100$ 。

题目分析及解答

普通版

直接暴力（$70\%$）

考虑到这道题大多数点的数据不大，可以直接暴力枚举所以客栈选择并进行判断，可以拿到大多数分数。

#include <iostream>
using namespace std;
int n,k,p,m[200300][2],cnt=0;
int main()
{
	cin>>n>>k>>p;
	for(int i=1;i<=n;i++)
	{
		cin>>m[i][0]>>m[i][1];//输入颜色与最低消费
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=i+1;j<=n;j++)
		{
			if(m[i][0]!=m[j][0]) continue;//如果颜色不同，不判断
			for(int k=i;k<=j;k++)
				if(m[k][1]<=p)//若存在任意一个咖啡店符合要求，答案加一
				{
					cnt++;
					break;
				}
		}
	}
	cout<<cnt;
	return 0;
}

优化暴力（$100\%$）

先前那种暴力会在判断颜色是否相同时耗过多时间。因此，我们可以分别存每种颜色所对应的客栈位置，再遍历两个相同颜色客栈之间咖啡馆的最低消费。（其实这样做仍旧可能过不了一些情况，比如说颜色全都相同时。但因为这道题的数据相当水，所以仍旧能够满分过，那些会被卡的点要用加强版代码解决。）

#include <iostream>
using namespace std;
int n,k,p,m[200300][2],c[51][200300],cnt=0;
int main()
{
	cin>>n>>k>>p;
	for(int i=1;i<=n;i++)
	{
		cin>>m[i][0]>>m[i][1];
		c[m[i][0]][++c[m[i][0]][0]]=i;//录入对应颜色数组
	}
	for(int i=0;i<k;i++)//遍历颜色
	{
		for(int j=1;j<=c[i][0];j++)
		{
			for(int l=j+1;l<=c[i][0];l++)
				for(int f=c[i][j];f<=c[i][l];f++)
					if(m[f][1]<=p)
					{
						cnt++;
						break;
					}
		}
	}
	cout<<cnt;
	return 0;
}

加强版

分析

对于每个点，如果自身消费小于最大值，那么它就可以使自身以及之前的所有点（不管颜色是否相同）与其之后的同色点组合。因此，我们可以对于每个点进行判断，如果消费小于最大值，就使自身及自身前未被标记的点被被标记，并将答案加上同色被标记点数目减一（不能和自身组合），否则将答案加同色被标记点数。

栗子

以本题样例为栗：

对于点一，消费大于最大值，不标记，答案加同色标记点数量$=0$；
对于点二，消费等于最大值，标记自身与点一，答案加同色标记点数量$-1=0$；
对于点三，消费小于最大值，标记自身，答案加同色标记点数量$-1=1$；
对于点四，消费大于最大值，不标记，答案加同色标记点数量$=2$；
对于点五，消费大于最大值，不标记，答案加同色标记点数量$=3$；

代码实现

#include <iostream>
using namespace std;
long long n,k,p,sum[10050],cnt=0;
struct node{
	int c,v;
}m[2000400];
int main()
{
	cin>>n>>k>>p;
	for(int i=1;i<=n;i++)
	{
		cin>>m[i].c>>m[i].v;
	}
	int j=0;//一开始标记到第0个点
	for(int i=1;i<=n;i++)
	{
		if(m[i].v<=p)
		{
			while(j<i)
			{
				j++;//更新目前标记到的点编号
				sum[m[j].c]++;
			}
			cnt+=sum[m[i].c]-1;//注意：自己不能和自己组合，故要减一
		}
		else cnt+=sum[m[i].c];//这里因为没有标记自己而不用考虑和自己组合的情况
	}
	cout<<cnt;
	return 0;
}

注意事项

对于加强版，数据组数与答案大小会变大，记得改数组大小并开$long$ $long$。