题意大致是:给你一个n (小于 3e5),表示字串长度,再把这个字符串 s 给你。要求把这个字符串分成四份,第四份必须是一个回文串,第一份的正序和第三份的逆序必须相等,求第二份的最短长度,每一份都可以是一个空字符串。
思路大致是,用字符串哈希判断第四串是不是回文,如果是,就二分的查找前面的一三串,时间复杂度O(NlogN) 。
#include <bits/stdc++.h>
typedef long long ll;
typedef long long ull;
using namespace std;
#define pb push_back
#define inf 0x3f3f3f3f
const int maxn=3e5+10;
const ull mod[]={998244353, 1000000007};
int n;
string s;
ull pre[2][maxn];
ull suf[2][maxn];
ull po[2][maxn];
pair<ull,ull> get1(int l,int r){
if (l>r) return {};
ull has0=(pre[0][r]-pre[0][l-1]*po[0][r-l+1])%mod[0];
has0=(has0+mod[0])%mod[0]; //防止是负数
ull has1=(pre[1][r]-pre[1][l-1]*po[1][r-l+1])%mod[1];
has1 =(has1 +mod[1])%mod[1];
return {has0,has1};
}
pair<ull,ull> get2(int l,int r){
if (l>r) return {};
ull has0=(suf[0][l]-suf[0][r+1]*po[0][r-l+1])%mod[0];
has0=(has0+mod[0])%mod[0];
ull has1=(suf[1][l]-suf[1][r+1]*po[1][r-l+1])%mod[1];
has1=(has1+mod[1])%mod[1];
return {has0,has1};
}
int main(){
memset(po,0,sizeof(po));
memset(suf,0,sizeof(suf));
memset(pre,0,sizeof(pre));
cin>>n;
cin>>s;
s=" "+s;
for (int j=0;j<=1;j++){
po[j][0]=1;
for (int i=1;i<=n;i++){
pre[j][i]=(pre[j][i-1]*1331+s[i])%mod[j];
po[j][i]=(po[j][i-1]*1331)%mod[j];
}
for (int i=n;i>=1;i--){
suf[j][i]=(suf[j][i+1]*1331+s[i])%mod[j];
}
}
int ans=inf;
for (int i=1;i<=n+1;i++){ //注意是n+1,因为第四串可能是空串。
auto mid1=get1(i,n);
auto mid2=get2(i,n);
int temp=0;
if (!(mid1==mid2)) continue;
int l=1,r=(i-1)/2+1;
while (l<r){
int mid=(l+r)>>1;
auto has1=get1(1,mid);
auto has2=get2(i-mid,i-1);
if (has1==has2){
temp=mid;
l=mid+1;
}
else r=mid;
}
ans=min(ans,i-1-2*temp);
}
cout<<ans<<endl;
}