CodeForces1345B. Ternary String

給一個只含有‘1’，‘2’，‘3’的字符串，找到同時含有‘1’，‘2’，‘3’的一個長度最小的子串。

設置一個數組$a[4][N]$，$a[j][i]$表示字符串第$i$位的字符，前面離它最近的字符‘j’的下標。以$a[2][10]$爲例，就是在字符串第10位之前，最靠近第10位的字符‘2’的下標。

$a[s[i]-'0'][i]=i$，距離當前位最近的$s[i]-'0'$字符當然是自己~

以$i$爲終點且滿足條件的子串最短長度爲$i-min(a[1][i],a[2][i],a[3][i])+1$

注意是min，因爲需要‘1’,’2‘,’3’都包含

初始化都初始化爲$-INF$,這樣前面不存在一個字符時，就會讓$i-(-INF)$,就不會更新答案了。

#include <bits/stdc++.h>
#define pb push_back 
#define fir first
#define sec second
#define ms(a,b) memset(a,b,sizeof(a)) 
#define INF 0x3f3f3f3f
#define sp system("pause")
using namespace std;
typedef long long ll;
typedef double db;
const int N=2e5+5;
const int mod=10007;
const db pi=acos(-1.0);
int a[4][N];
int main()
{
    int t;
    string s;
    cin>>t;
    while(t--)
    {
        cin>>s;
        int l=s.size(),minn=INF;
        a[1][0]=a[2][0]=a[3][0]=-INF;//初始化
        a[s[0]-'0'][0]=0;
        for(int i=1;i<l;i++)
        {
            int mini;
            for(int j=1;j<=3;j++) a[j][i]=a[j][i-1];
            a[s[i]-'0'][i]=i;
            if(s[i]=='1')  mini=max(i-a[2][i],i-a[3][i])+1;
            else if(s[i]=='2') mini=max(i-a[1][i],i-a[3][i])+1;
            else mini=max(i-a[1][i],i-a[2][i])+1;
            minn=min(minn,mini);//更新最小值
        }
        if(minn==INF) cout<<0<<endl;
        else cout<<minn<<endl;
    }
    
    #ifndef ONLINE_JUDGE
    sp;
    #endif
}