URAL1297 Palindrome 后缀数组

http://fastvj.rainng.com/problem/URAL-1297

题意为求最长回文串。

这个题也好做,我们就把输入的字符串str和他自己反向链接,然后求sa数组和height数组;

最后遍历后缀数组,这个时候就要细心了,首先你要判断sa[i-1],sa[i]是不是分别属于两个字符串了,并且还有判断是不是回文串的起点,记录位置的变量初始化为0,就可以了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1000010;
int s[N],x[N],y[N],n,m;
int c[N],sa[N],height[N],rk[N];
void Suffix()
{
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;x[sa[0]]=0;
        for(int i=1;i<n;i++)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p>=n) break;
        m=p;
    }
}
void geth()
{
    for(int i=0;i<n;i++) rk[sa[i]]=i;
    for(int i=0,k=0;i<n;i++)
    {
        if(rk[i])
        {
            if(k) --k;
            int j=sa[rk[i]-1];
            while(s[i+k]==s[j+k])
                k++;
            height[rk[i]]=k;
        }
    }
}
int k;
char str1[N],str2[N];

int main()
{

    while(~scanf("%s",str1))
    {
        n=0;
        int l=strlen(str1);
        for(int i=0;i<l;i++) s[n++]=str1[i]-'A'+1;
        s[n++]=127;
        for(int i=l-1;i>=0;i--) s[n++]=str1[i]-'A'+1;
        s[n++]=0;
        m=128;
        Suffix();
        geth();
        int ans=1,pos=0;
        for(int i=1;i<n;i++)
        {
            int mx=max(sa[i],sa[i-1]);
            int mn=min(sa[i],sa[i-1]);
            if(mx<l||mn>l) continue;
            if(mn+height[i]!=n-1-mx) continue;
            if(height[i]>ans)
            {
                ans=height[i];
                pos=mn;
            }
            if(height[i]==ans) pos=min(mn,pos);
        }
        for(int i=pos;i<pos+ans;i++) printf("%c",s[i]+'A'-1);
        printf("\n");
    }
    return 0;
}

 

上一篇:Python:pathlib模块


下一篇:Camel框架的使用