洛谷 P1311 [NOIP2011 提高组] 选择客栈题解

原题目：

P1311 [NOIP2011 提高组] 选择客栈 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

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

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

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

输入格式

共 n+1 行。

第一行三个整数 n,k,p，每两个整数之间用一个空格隔开，分别表示客栈的个数，色调的数目和能接受的最低消费的最高值；

接下来的 n 行，第 i+1 行两个整数，之间用一个空格隔开，分别表示 i 号客栈的装饰色调 ai​ 和 i 号客栈的咖啡店的最低消费 bi​。

输出格式

一个整数，表示可选的住宿方案的总数。

100分递推思路

因为每一家客栈也是一家咖啡店，只要每一个点 i 都向前循环到 j 判断就能避免重复，判断条件为ans += (a[i] == a[j] ? 1:0)。满足条件的话就分类讨论

①若b[i] <= p时，则不用判断中间是否有值<=p，遇到颜色相同的话ans就+1

②若b[i] > p时，则判断中间是否有值<=p，若有的话，此后的ans才能+1

但是每个点都循环到第一个点的话会超出时间，我们发现中间有很多重复相同的步骤，每个点都在判断颜色是否与当前点相同，那么我们用一个标记数组fa[i]来记录第i种颜色符合的个数

<----------------------------------->分界线

①若b[i] > p时，ans就只用加上fa[a[i]]的个数

②若b[i] <= p时，说明以后的点若想要再找与自己颜色相同的点的话，ans就只用加上f[a[i]]的个数，所以我们要将前面的点从 j 开始都打上标记，并且加上f[a[i]]的个数-1

实例：

当 i 遍历到：

1号：(b[1] > p) 加上与自己同样颜色的点的个数；j=1；ans=0

2号：(b[2] <= p) 将1号点的fa[1]+1 和 自己2号点的fa[0]+1；j=2；ans=0

3号：(b[3] > p) 加上与自己同样颜色的点的个数；j=2；ans=0

4号：(b[4] <= p) 将3号点的fa[2]+1 和 自己4号点的fa[1]+1；j=4；ans=1

5号：(b[5] <= p) 将 自己4号点的fa[0]+1；j=5；ans=2

6号：(b[6] > p) 加上与自己同样颜色的点的个数；j=5；ans=3

完整代码

#include <bits/stdc++.h>
using namespace std;
int n,m,p;
long long ans;
int a[2000005];
long long b[200005],fa[2000005];
int main ()
{
	int i,j=0;
	cin>>n>>m>>p;
	for (i=1;i<=n;i++)
	{
		cin>>a[i]>>b[i];
	}
	for (i=1;i<=n;i++)
	{
		if (b[i]>p) ans+=fa[a[i]];
		else
		{
			while(j<i)
			{
				j++;
				fa[a[j]]++;
			}
			ans+=fa[a[i]]-1;
		}
	}
	cout<<ans;
	return 0;
}

最后成功AC