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;
}

你可能感兴趣的:(CodeForces 616 D. Longest k-Good Segment(尺取))