CodeForces616D Longest k-Good Segment

题意

找序列中最长的包括不超过k个不同元素的子区间，输出任意合法区间的左右端点下标。

思路

尺取法，用数组存储元素的出现次数。

代码

#include <stdio.h>
#include <string.h>
int n,k,v[1000001],a[500001];
int l,r,mid;
int check(int m)
{
	memset(v,0,sizeof(v));
	int sum=0;
	for (int i=1;i<=m;++i)
	{
		if (!v[a[i]]) sum++;
		v[a[i]]++;
	}
	if (sum<=k) return 1;
	for (int i=m+1;i<=n;++i)
	{
		v[a[i-m]]--;
		if (!v[a[i-m]]) sum--;
		if (!v[a[i]]) sum++; 
		v[a[i]]++;
		if (sum<=k) return 1;
	}	
	return 0;
}
int main()
{
	scanf("%d%d",&n,&k);
	for (int i=1;i<=n;++i) scanf("%d",&a[i]);
	if (n<=k) return printf("1 %d\n",n),0;
	for (l=k,r=n;l<r;)
	{
		mid=((l+r+1)>>1);
		if (check(mid)) l=mid;
		else r=mid-1;
	}
	int sum=0;
	mid=l;
	memset(v,0,sizeof(v));
	for (int i=1;i<=mid;++i)
	{
		if (!v[a[i]]) sum++;
		v[a[i]]++;
	}
	if (sum<=k) return printf("%d %d\n",1,mid),0;
	for (int i=mid+1;i<=n;++i)
	{
		v[a[i-mid]]--;
		if (!v[a[i-mid]]) sum--;
		if (!v[a[i]]) sum++;
		v[a[i]]++;
		if (sum<=k) return printf("%d %d\n",i-mid+1,i),0;
	}
}