TYVJ P1063 数字串 Label:双指针 线性扫描

描述

给你一个长度为n的数字串，数字串里会包含1-m这些数字。如果连续的一段数字子串包含了1-m这些数字，则称这个数字字串为NUM串。你的任务是求出长度最短的NUM串是什么，只需要输出这个长度即可。
1<=n，m<=200000

输入格式

第一行给定n和m。 
第二行n个数，表示数字串，数字间用空格隔开。

输出格式

如果存在NUM串则输出最短NUM串长度，否则输出“NO”。

测试样例1

输入

5 3 
1 2 2 3 1

输出

3

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
int n,m,num[200005],a[200005],mn;
int cnt,l=0,r=1;//用两个指针，分别指向头和尾

void print(){
    for(int i=1;i<=m;i++) printf("%d ",num[i]);
    puts(" ");
}

int main(){
//    freopen("01.txt","r",stdin);
    scanf("%d%d",&n,&m);
    cnt=m;
    if(n<m){//n<m不可能有数字串
        cout<<"NO"<<endl;
        return 0;
    }
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);

    for(;r<=n;r++){//尾后移
        if(num[a[r]]==0) {
            ++num[a[r]];
            --cnt;
        }
        else ++num[a[r]];
        if(cnt==0) break;
    }
    if(cnt>0){
        cout<<"NO"<<endl;
        return 0;
    }
    
    mn=r-l;//l此时为0
    while(l<=n){//头后移
        ++l;
//        printf("l=%d r=%d
",l,r);
//        print();
        if(num[a[l]]==1) --num[a[l]],++cnt;
        else --num[a[l]];
//        cout<<cnt<<endl;
        if(cnt==0) {mn=min(mn,r-(l+1)+1);continue;}
        
        ++r;
        for(;r<=n;++r){
            if(num[a[r]]==0) ++num[a[r]],--cnt;
            else ++num[a[r]];
            if(cnt==0) break;
        }
        if(cnt) break;
        else mn=min(mn,r-(l+1)+1);
    }
    printf("%d
",mn);
    return 0;
}

cnt用于记录当前是否还需要添加元素