CodeForces1345B. Ternary String

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

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

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

i 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 imin(a[1][i],a[2][i],a[3][i])+1

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

初始化都初始化为 − I N F -INF INF,这样前面不存在一个字符时,就会让 i − ( − I N F ) i-(-INF) i(INF),就不会更新答案了。

#include 
#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

你可能感兴趣的:(思维)