非确定性有穷状态决策自动机练习题Vol.1 A.扭动的回文串
题目描述
(JYY)有两个长度均为(N)的字符串(A)和(B)。
一个“扭动字符串(S(i,j,k))由(A)中的第(i)个字符到第(j)个字符组成的子串
与B中的第(j)个字符到第(k)个字符组成的子串拼接而成。
比如，若(A)=’XYZ’，(B)=’UVW’，则扭动字符串(S(1,2,3))=’XYVW’。
(JYY)定义一个“扭动的回文串”为如下情况中的一个：
1.(A)中的一个回文串；
2.(B)中的一个回文串；
3.或者某一个回文的扭动字符串(S(i,j,k))
现在(JYY)希望找出最长的扭动回文串。
输入格式
第一行包含一个正整数(N)。
第二行包含一个长度为(N)的由大写字母组成的字符串A。
第三行包含一个长度为(N)的由大写字母组成的字符串B。
输出格式
输出的第一行一个整数，表示最长的扭动回文串。
样例
样例输入
5
ABCDE
BAECB
样例输出
5
样例解释
最佳方案中的扭动回文串如下所示（不在回文串中的字符用.表示）：
.BC..
..ECB
数据范围与提示
对于(10\%)的数据：(N≤100)；
对于(30\%)的数据：(N≤1000)；
对于(50\%)的数据：(N≤10000);
对于(100\%)的数据：(1≤N≤10^5)
分析
对于只在 (A) 中的回文串和只在 (B) 中的回文串，我们直接拿马拉车 (O(n)) 解决即可
比较难处理的是扭动回文串的情况
显然，对于每一个回文串，都有一个回文中心（在马拉车算法前，我们已经把所有的偶回文串填充成了奇回文串）
我们可以把 (A) 串和 (B) 串中的每一个字符都当作回文中心计算贡献
在马拉车算法后，我们已经求出了以 (i) 作为回文中心的最长回文半径 (f[i])
我们只需要在原来的基础上继续向外扩展即可
对于 (A) 串中的字符，我们在原串上向左扩展，在 (B) 串上向右扩展
对于 (B) 串中的字符，我们在原串上向右扩展，在 (A) 串上向左扩展
直接暴力去扫肯定会 (T) ，我们可以用二分+ (Hash) 将扩展的复杂度降为 (log(n))
对于 (A) 我们正着记录一遍哈希值，对于 (B) 我们倒着记录一遍哈希值即可
代码
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=1e6+5;
typedef unsigned long long ll;
const ll bas=233;
char a[maxn],b[maxn],ksa[maxn],ksb[maxn];
int fa[maxn],fb[maxn],n;
ll hasha[maxn],hashb[maxn],mi[maxn];
inline int read(){
int x=0,f=1;
char ch=getchar();
while(ch<'0' || ch>'9'){
if(ch=='-') f=-1;
ch=getchar();
}
while(ch>='0' && ch<='9'){
x=(x<<1)+(x<<3)+(ch^48);
ch=getchar();
}
return x*f;
}
int js=0;
ll get_hasha(int l,int r){
return hasha[r]-hasha[l-1]*mi[r-l+1];
}
ll get_hashb(int l,int r){
return hashb[l]-hashb[r+1]*mi[r-l+1];
}
//取哈希值
int solvea(int id){
int l=id-fa[id]+1,r=id+fa[id]-1;
int nl=0,nr=min(l-1,n-r),nmids;
while(nl<=nr){
nmids=(nl+nr)>>1;
if(get_hasha(l-nmids,l)==get_hashb(r-2,r+nmids-2)) nl=nmids+1;
else nr=nmids-1;
}
return fa[id]-1+nr;
}
int solveb(int id){
int l=id-fb[id]+1,r=id+fb[id]-1;
int nl=0,nr=min(l-1,n-r),nmids;
while(nl<=nr){
nmids=(nl+nr)>>1;
if(get_hasha(l-nmids+2,l+2)==get_hashb(r,r+nmids)) nl=nmids+1;
else nr=nmids-1;
}
return fb[id]-1+nr;
}
//对于两个串分别二分
int main(){
n=read();
scanf("%s%s",ksa+1,ksb+1);
n=n*2+1;
a[0]='$',b[0]='$';
for(int i=1;i<=n;i++){
if(i&1) a[i]='#';
else a[i]=ksa[i/2];
}
for(int i=1;i<=n;i++){
if(i&1) b[i]='#';
else b[i]=ksb[i/2];
}
mi[0]=1;
for(int i=1;i<=n;i++){
mi[i]=mi[i-1]*bas;
hasha[i]=hasha[i-1]*bas+a[i];
}
for(int i=n;i>=1;i--){
hashb[i]=hashb[i+1]*bas+b[i];
}
//预处理哈希值
int mids=0,r=0,ans=0;
for(int i=1;i<=n;i++){
if(i<=r) fa[i]=min(fa[mids*2-i],r-i+1);
while(a[i+fa[i]]==a[i-fa[i]]) fa[i]++;
if(i+fa[i]-1>r){
r=i+fa[i]-1,mids=i;
ans=max(ans,fa[i]-1);
}
}
mids=0,r=0;
for(int i=1;i<=n;i++){
if(i<=r) fb[i]=min(fb[mids*2-i],r-i+1);
while(b[i+fb[i]]==b[i-fb[i]]) fb[i]++;
if(i+fb[i]-1>r){
r=i+fb[i]-1,mids=i;
ans=max(ans,fb[i]-1);
}
}
//马拉车
for(int i=1;i<=n;i++){
ans=max(ans,solvea(i));
ans=max(ans,solveb(i));
}
printf("%d
",ans);
return 0;
}