[NOIp2011] luogu P1311 选择客栈

我妈的抽象歌曲真 nb。

题目描述

给你 $n$ 个点，每个点有两个参数 $c_i,d_i$，给你一个数 $D$。定义一种方案合法，当且仅当你选出整数 $i,j\in[1,n],i<j,c_i=c_j$，且存在一个整数 $k\in(i,j)$ 使得 $d_k\leq D$。求合法方案数。

Solution 1

这是我口胡的做法。

注意到 $c$ 特别小。每次枚举一个 $c$，然后枚举一个 $c_k\leq D$，如果 $k$ 前面有 $x$ 个 $c_x=c$， 后面有 $y$ 个 $c_y=c$，那么就把答案加 $xy$。时间复杂度 $O(nk)$。

还有一种思路就是你看到 $p$ 其实也很小，然后还是用类似的方法做，可能更优吧，我也没怎么想。能卡过。

Solution 2 (Idea From ShawnZhou)

#include<cstdio>
#include<cstdlib>
#include<cstring>

const int MAXC=60;

int n,k,p;
int c,d;
int now;
int last[MAXC];	//上一个相同颜色的点的位置
int sum[MAXC];
int cnt[MAXC];
int ans=0;

int main(){
	scanf("%d%d%d",&n,&k,&p);
	for(int i=1;i<=n;++i){
		scanf("%d%d",&c,&d);
		if(d<=p) now=i;
		if(now>=last[c])
			sum[c]=cnt[c];
		last[c]=i;
		ans+=sum[c];
		++cnt[c];
	}
	printf("%d",ans);
}