P1638 逛画展 (尺取法)

P1638 逛画展 (尺取法)

传送门

思路:显然区间具有单调性,所以可以通过枚举区间左端点,让区间右端点不断增加,至于区间画的种类数,用一个数组记录区间的[l , r]的画的种类数,不断更新即可。

时间复杂度: O ( n ) O(n)

AC代码:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=1e6+5;
#define mst(a) memset(a,0,sizeof a)
int a[N],cnt[N],n,m,x,y=1e9+7;//用cnt[i]维护区间每种画的个数. 
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int l=1,r=0,sum=0;r<=n;){
		while(r<n&&sum<m){
			if(!cnt[a[++r]]) sum++;
			cnt[a[r]]++;
		}
		if(sum<m) break;
		if(r-l+1<y-x+1){//因为sum<m会break 所以此处sum>=m 只需判断让区间长度最小化即可. 
			x=l,y=r;
		}
		if(--cnt[a[l++]]==0) sum--;//更新区间左端点. 
	}
	printf("%d %d\n",x,y);	
	return 0;
}
原创文章 201 获赞 165 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_45750972/article/details/106114159