1. 程式人生 > 其它 >字串雜湊例項---迴文串

字串雜湊例項---迴文串

迴文子串的最大長度

技巧

在每兩個字元中插入一個新的字元,可以不用判斷字串的奇偶,方便程式設計。

利用字串雜湊和二分求解

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 2000010;
const int P = 131;

typedef unsigned long long Ull;

//h1正序,h2逆序
Ull h1[N],h2[N],p[N];
int n;
char s[N];

Ull get(Ull h[],int l,int r){
    return h[r] - h[l - 1] * p[r - l + 1];
}

int main(){
    int t = 1;

    while(cin >> s + 1,strcmp(s + 1,"END")){
        n = strlen(s + 1);

        //將所有的字串變成奇數串,方便計算
        //做法是在兩個字母中間插入一個符號
        for(int i = n * 2;i > 0;i -= 2){
            s[i] = s[i / 2];
            s[i - 1] = 'z' + 1;
        } 
        n *= 2;

        //獲取雜湊值
        p[0] = 1;
        for(int i = 1,j = n;i <= n;i ++,j --){
            h1[i] = h1[i - 1] * P + s[i] - 'a' + 1;
            h2[i] = h2[i - 1] * P + s[j] - 'a' + 1;
            p[i] = p[i - 1] * P;
        }

        int res = 0;
        for(int i = 1;i <= n;i ++){
            int l = 0,r = min(n - i,i - 1);     //二分法求迴文半徑

            while(l < r){
                int mid = l + r + 1 >> 1;

                if(get(h1,i - mid,i - 1) == get(h2,n - (i + mid) + 1,n - (i + 1) + 1)) l = mid;
                else r = mid - 1;
            }

            //看回文串有無包含插入的符號
            if(s[i - l] <= 'z') res = max(res,l + 1);
            else res = max(res,l);
        }

        printf("Case %d: %d\n",t ++,res);
    }
}