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[],a[],mn;
 int cnt,l=,r=;//用两个指针，分别指向头和尾

 void print(){
     for(int i=;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 ;
     }
     for(int i=;i<=n;i++)
         scanf("%d",&a[i]);

     for(;r<=n;r++){//尾后移
         if(num[a[r]]==) {
             ++num[a[r]];
             --cnt;
         }
         else ++num[a[r]];
         if(cnt==) break;
     }
     if(cnt>){
         cout<<"NO"<<endl;
         return ;
     }

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

         ++r;
         for(;r<=n;++r){
             if(num[a[r]]==) ++num[a[r]],--cnt;
             else ++num[a[r]];
             if(cnt==) break;
         }
         if(cnt) break;
         else mn=min(mn,r-(l+)+);
     }
     printf("%d\n",mn);
     return ;
 }

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