CodeForces 616 D. Longest k-Good Segment（尺取）

Description

给出一个长度为n的序列，如果这个序列的某一个子段含有不超过k个不同的数字则称这个子段是第k好的子段，问这个序列最长的第k好子段的起始位置和结束位置

Input

第一行为两个整数n和k，之后为n个整数ai表示该序列(1<=k<=n<=5*10^5,0<=ai<=10^6)

Output

这个序列最长的第k好子段的起始位置和终止位置

Sample Input

9 3

6 5 1 2 3 2 1 4 5

Sample Output

3 7

Solution

尺取，记录[l,r]中不同数的个数，每次这个数小于k那么r++，如果已经大于k则l++直到个数不大于k，每次更新完r之后判断r-l+1是否比已有答案更优来更新答案即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define maxn 555555
int n,k,cnt,a[maxn],flag[2*maxn];
int main()
{
while(~scanf("%d%d",&n,&k))
{
memset(flag,0,sizeof(flag));cnt=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(i<=k)
{
if(!flag[a[i]])cnt++,flag[a[i]]=1;
else flag[a[i]]++;
}
}
int l=1,r=k,ansl=1,ansr=k;
while(r<n)
{
if(flag[a[r+1]]||cnt<k)
{
r++;
if(flag[a[r]]==0)cnt++;
flag[a[r]]++;
if(ansr-ansl<r-l)ansl=l,ansr=r;
}
else
{
if(flag[a[l]]==1)cnt--;
flag[a[l]]--;
l++;
}
}
printf("%d %d\n",ansl,ansr);
}
return 0;
}