CodeForces1345B. Ternary String

给一个只含有‘1’,‘2’,‘3’的字符串,找到同时含有‘1’,‘2’,‘3’的一个长度最小的子串。

设置一个数组 a [ 4 ] [ N ] a[4][N] a [ j ] [ i ] a[j][i] 表示字符串第 i i 位的字符,前面离它最近的字符‘j’的下标。以 a [ 2 ] [ 10 ] a[2][10] 为例,就是在字符串第10位之前,最靠近第10位的字符‘2’的下标。

a [ s [ i ] 0 ] [ i ] = i a[s[i]-'0'][i]=i ,距离当前位最近的 s [ i ] 0 s[i]-'0' 字符当然是自己~

i i 为终点且满足条件的子串最短长度为 i m i n ( a [ 1 ] [ i ] , a [ 2 ] [ i ] , a [ 3 ] [ i ] ) + 1 i-min(a[1][i],a[2][i],a[3][i])+1

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

初始化都初始化为 I N F -INF ,这样前面不存在一个字符时,就会让 i ( I N F ) 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
}

猜你喜欢

转载自blog.csdn.net/Luowaterbi/article/details/106219388
今日推荐