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

最新推荐文章于 2020-12-05 21:22:14 发布

v5zsq

最新推荐文章于 2020-12-05 21:22:14 发布

阅读量863

点赞数

CC 4.0 BY-SA版权

分类专栏： Code Forces 尺取

本文链接：https://blog.youkuaiyun.com/V5ZSQ/article/details/50750827

Code Forces 同时被 2 个专栏收录

606 篇文章

订阅专栏

尺取

19 篇文章

订阅专栏

本文介绍了一种通过尺取法解决寻找序列中最长第k好子段问题的算法实现，尺取法通过移动左右边界来调整子段内不同数字的数量，以确保其不超过k个，从而找到满足条件的最长子段。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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