给一个只含有‘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 i−min(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 <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
}