【NOIP 统计】JZOJ_3025 选择客栈

题意

有$n$间客栈，它们有$k$种色调，其中每间客栈有一个最低花费为$v_i$的咖啡店。两人住在两间色调相同的房间，他们希望在他们房间中（包括这两间）有最低花费不超过$p$的咖啡店。求出他们的住房方案数。

思路

先统计一个前缀和，记录每种颜色的房间有多少个。
然后从后往前枚举一个端点，再枚举咖啡店，如果满足条件就加上前面相同色调的房子个数。
可以有小小的剪枝。

代码

#include<cstdio>
#include<algorithm>

int n, k, p, ans;
int hotel[200001][3], pre[200001][51];

int main() {
	scanf("%d %d %d", &n, &k, &p);
	for (int i = 1; i <= n; i++) {
		scanf("%d %d", &hotel[i][1], &hotel[i][2]);
		for (int j = 0; j < k; j++)
			pre[i][j] = pre[i - 1][j] + 1 * (hotel[i][1] == j);
	}
	for (int i = n; i >= 1; i--) {
		if (!pre[i - 1][hotel[i][1]]) continue;
		for (int j = i; j >= 1; j--) {
			if (hotel[j][2] <= p) {
				if (j != i) ans += pre[j][hotel[i][1]];
				else ans += pre[j - 1][hotel[i][1]];
				break;
			}
			if (!pre[j][hotel[i][1]]) break;
		}
	}	
	printf("%d", ans);
}