Codeforces Global Round 7 D2. Prefix-Suffix Palindrome (Hard version)(Manacher算法)

题意:

取一字符串前缀和后缀(均可为空)构成最长回文串。

思路:

先同时取对称的前后缀,之后再取较长的前缀或后缀。

#include <bits/stdc++.h>
using namespace std;

string Manacher(const string &s){
    string t="#";
    for(char c:s){
        t+=c;
        t+="#";
    }

    int RL[t.size()]={0};
    int MaxRight=0;
    int pos=0;

    for(int i=0;i<t.size();i++){
        if(i<MaxRight)
            RL[i]=min(RL[2*pos-i],MaxRight-i);
        else
            RL[i]=1;
        while(i-RL[i]>=0&&i+RL[i]<t.size()&&t[i-RL[i]]==t[i+RL[i]])
            RL[i]+=1;
        if(RL[i]+i-1>MaxRight)
            pos=i,MaxRight=RL[i]+i-1;
    }

    int MaxLen=0;
    for(int i=0;i<t.size();i++)
        if(RL[i]==i+1)
            MaxLen=i;
    return s.substr(0,MaxLen);
}

void solve(){
    string s;cin>>s;

    int l=0,r=s.size()-1;
    vector<char> lf,rt;

    while(l<=r&&s[l]==s[r]){
        lf.push_back(s[l]);
        rt.push_back(s[r]);
        ++l,--r;
    }
    if(l>=r){
        cout<<s<<"\n";
        return;
    }

    string s1=s.substr(l,r-l+1);
    string s2=s1;
    reverse(s2.begin(),s2.end());

    s1=Manacher(s1);
    s2=Manacher(s2);

    reverse(rt.begin(),rt.end());
    cout<<lf;
    cout<<(s1.size()>s2.size()?s1:s2);
    for(char c:rt) cout<<c;
    cout<<"\n";
}

int main()
{
    int t;cin>>t;
    while(t--)
        solve();
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/Kanoon/p/12530330.html