LeetCode 214. 最短回文串--Rabin-Karp 字符串编码+双重编码判定

214. 最短回文串
     给定一个字符串 s，你可以通过在字符串前面添加字符将其转换为回文串。找到并返回可以用这种方式转换的最短回文串。

示例 1：

输入：s = “aacecaaa”
输出：“aaacecaaa”
示例 2：

输入：s = “abcd”
输出：“dcbabcd”

提示：

0 <= s.length <= 5 * 104
s 仅由小写英文字母组成

题解

很巧妙的题目，先了解下Rabin-Karp编码，那么题目就很简单了，我们从左往右遍历，通过这个编码得到两个数字ans1和ans2，ans1表示[0,i]之间字符串构成的数字，ans2表示[i+1,2i+1]之间构成的字符串表示的数字，如果ans1==ans2，那么就意味着[0,i]与[i+1,2i+1]可以构成回文串，那么以i位置打开，往字符串前面添加字符，那么答案就出来了，因为这个编码得到的数字很大所以需要取模，又因为取模后导致两个不同的字符串得到的数字会相同，于是我们用两个取模方式，双重判定，得到的答案就是唯一了，这里还要判断构成的是奇数回文串，
于是ans3表示[i+2,2*i+2]之间字符串构成的数字。答案就搞定了。

AC代码

class Solution {
public:
    typedef long long ll;
    ll mod_ans=100000000000000+7;
    ll mod_sum=100000000000+7;
    bool check(string s)
    {
        for(int i=0;i<s.length()/2;i++)
        if(s[i]!=s[s.length()-1-i])return false;
        return true;
    }
    string shortestPalindrome(string s) 
    {
        if(s=="")return "";
        if(check(s))return s;
        ll ans1=0,ans2=0,ans3=0;
        ll base_ans=1;
        ans1=(s[0]-'a');
        ans2=(s[1]-'a');
        ans3=(s[2]-'a');

        ll sum1=0,sum2=0,sum3=0;
        ll base_sum=1;
        sum1=(s[0]-'a');
        sum2=(s[1]-'a');
        sum3=(s[2]-'a');
        int d1=0,d2=0;
        for(int i=1;i<=s.length()/2;i++)
        {
            if(ans1==ans2&&sum1==sum2)
            d1=i;
            if(s.length()%2==0||i<s.length())
            {
                if(ans1==ans3&&sum1==sum3)
                d2=i;                
            }
            
            if(i==s.length()/2)break;

            ans1+=((s[i]-'a')*(base_ans*26));
            ans1%=mod_ans;

            ans2=(ans2+mod_ans-(s[i]-'a')*base_ans);
            ans2*=26;
            ans2+=(s[2*i]-'a');
            ans2*=26;
            ans2+=(s[2*i+1]-'a');
            ans2%=mod_ans;
            if(s.length()%2==0||i<s.length())
            {
                ans3=(ans3+mod_ans-(s[i+1]-'a')*base_ans);
                ans3*=26;
                ans3+=(s[2*i+1]-'a');
                ans3*=26;
                ans3+=(s[2*i+2]-'a');
                ans3%=mod_ans;
            }

            base_ans*=26;
            base_ans%=mod_ans;

            sum1+=((s[i]-'a')*(base_sum*26));
            sum1%=mod_sum;

            sum2=(sum2+mod_sum-(s[i]-'a')*base_sum);
            sum2*=26;
            sum2+=(s[2*i]-'a');
            sum2*=26;
            sum2+=(s[2*i+1]-'a');
            sum2%=mod_sum;

            if(s.length()%2==0||i<s.length())
            {
                sum3=(sum3+mod_sum-(s[i+1]-'a')*base_sum);
                sum3*=26;
                sum3+=(s[2*i+1]-'a');
                sum3*=26;
                sum3+=(s[2*i+2]-'a');
                sum3%=mod_sum;
            }

            base_sum*=26;
            base_sum%=mod_sum;
           // cout<<0<<" "<<i<<" "<<ans1<<endl;
          //  cout<<i+1<<" "<<2*i+1<<" "<<ans2<<endl;
          //  cout<<i+2<<" "<<2*i+2<<" "<<ans3<<endl;
        }
        string res="";
        string t="";
        if(d2>=d1)
        {
            for(int i=s.length()-1;i>d2;i--)
            t+=s[i];
            res=t;
            res+=(s[d2]);
            reverse(t.begin(),t.end());
            res+=t;
        }
        else
        {
            for(int i=s.length()-1;i>=d1;i--)
            t+=s[i];
            res=t;
            reverse(t.begin(),t.end());
            res+=t;
        }
        return res;
    }
};

居然速度这么快，开心~