首先,通过向原字符串每两个字符中间添加一些特殊无效字符‘#’将其转化成长度为奇数的串,简化代码复杂度,把奇回文和偶回文统一处理。
同时,在原字符串开头加上一个‘$’,省去越界处理,因为开头的‘$’和末尾的‘\0’一定不相等,所以回文扩展时遇到不相等的字符会自动停止。
然后设置一个p数组,p[i]表示以i为回文中心点的最长回文半径长度,显然此时最长回文串的长度=p[i]-1(因为每两个字符中间都插入了一个‘#’),回文串的起点=i-p[i]-1,终点=i+p[i]-1.
设置两个指针id,mx,分别表示当前最大回文串(扩展到的右端点最远的回文串)的中心点和右端点,下面分两种情况讨论:
当i<mx时,p[i]=min(p[2*id-i],mx-i),i的能判断出的回文的右端点最大为mx,所以当前p[i]最大为mx-i,通过中点坐标公式:id=(i+j)/2 (j为与i关于id对称的左边那个点),所以j=2*id-i,因为p[j]在这之前已经求出,又因为i和j在当前id为中心的回文串内且i和j关于id对称,所以j的最大回文半径即为i的在id的回文串内的最大回文半径,然后超出的部分就要再一一比较了。
当i>=mx时,p[i]=1,i超出了当前以id为中心的回文串的范围,所以从p[i]=1开始只能一一比较。
接下来就是常规的字符一一比较然后扩展回文串长度部分。
注意:在这个过程中id要不断随着mx的更新而更新。
最后,遍历p数组即可得到最大回文半径及其对应的起点,从而得到最长回文子串。
时间复杂度分析:因为mx在遍历字符串的长度过程中不断增加,是一个线性的过程,通过p数组,省去了重复的比较,就是遍历一个字符串的长度的次数,所以时间复杂度为o(n)。
完整代码:
#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
using namespace std;
const int maxn=1e5+5;
int p[maxn<<1];//注:因为加了一些奇奇怪怪的符号,所以p数组大小要为原串长度的两倍
//注:因为每次都是按顺序从前往后更新p数组,所以p数组不用初始化
void init(string &s)
{
string m="$#";
for(int i=0;i<s.size();i++){
m+=s[i];
m+='#';
}
s=m;
}
string manacher(string s)
{
init(s);
//cout<<"ins="<<s<<endl;
int id=1,mx=1;
for(int i=1;i<s.size();i++){
if(i<mx) p[i]=min(p[2*id-i],mx-i);
else p[i]=1;
while(s[i-p[i]]==s[i+p[i]]) p[i]++;
if(mx<i+p[i]){
mx=i+p[i];
id=i;
}
}
//找最大回文长度:
int maxlen=0;
for(int i=1;i<s.size();i++){
if(p[i]>maxlen){
maxlen=p[i];
id=i;
}
}
maxlen--;
//构造原最长回文串:
int start=id-maxlen;
int ed=id+maxlen;
string ans="";
for(int i=start;i<=ed;i++){
if(s[i]!='#') ans+=s[i];
}
return ans;
}
int main()
{
string str;
cin>>str;
cout<<manacher(str)<<endl;
return 0;
}