[manacher] CF1326D2

题目

题目链接：https://codeforces.com/contest/1326/problem/D2

思路

有两种情况，
第一种情况是前后字串有一个是空，跑这个时候的manacher,询下这个时候的最长回文串。
第二种情况是前后没有空串，把前后的回文串记录下，对剩下中间的跑一次manacher

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
int s1,s2;
int p[3000010];
void manacher(string ss){
	string sn;
	sn.push_back('$');
	sn.push_back('#');
	for(int i=0;i<=ss.size()*2+5;i++) p[i]=0;
	for(int i=1;i<=(int)ss.size();i++){
		sn.push_back(ss[i-1]);
		sn.push_back('#');
	}
	sn.push_back('\0');
//	cout<<sn<<endl;
	int mid;
	int mx=0;
	for(int i=1;i<=(int)sn.size();i++){
		if(i<mx){
			p[i]=min(p[2*mid-i],mx-i);
		}
		else p[i]=1;
		while(sn[i-p[i]]==sn[i+p[i]]) p[i]++;
		if(mx-i<p[i]){
			mid=i;
			mx=p[i]+i;
		}
		int num=p[i]-1;
		if(i%2==0){
			num=(num-1)/2;
			int p1=i/2;
			if(p1-num==1) s1=max(s1,p[i]-1);
			if(p1+num==ss.size()) s2=max(s2,p[i]-1);
		}
		else{
			num=num/2;
			int p1=(i-1)/2,q=(i+1)/2;
			if(p1-num+1==1) s1=max(s1,p[i]-1);
			if(q+num-1==ss.size()) s2=max(s2,p[i]-1);
		}
	}
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int t;
	cin>>t;
	while(t--){
		string res1,res2,res3,s,ss;
		cin>>s;
		s1=0,s2=0;
		manacher(s);
		if(s1>s2) res1=s.substr(0,s1);
		else res1=s.substr(s.size()-s2);
		int l=0,r=s.size()-1;
		while(r>l&&s[l]==s[r]) l++,r--;
		res2=s.substr(0,l);
		if(r>l){
			ss=s.substr(l,r-l+1);
			s1=0,s2=0;
			manacher(ss);
			if(s1>s2) {
				res2+=ss.substr(0,s1);
				}
			else res2+=ss.substr(ss.size()-s2);}
		res2+=s.substr(r+1);
		cout<<(res1.size()>res2.size()?res1:res2)<<endl;
	} 
    return 0;
}