manacher算法学习心得

详细解释见：https://segmentfault.com/a/1190000008484167

manacher算法，算法处理回文字符串。

通过初始化，在s串前加'$'在各个字符之间加'#'构成一个新的奇长度的串，对于这个串，进行预处理，构造p数组，其中p[i]-1等于回文串长度，处理方法见代码：

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

using namespace std;

char s[1000];
char s_new[2000];
int p[2000];

int Init()
{
    int len = strlen(s);
    s_new[0] = '$';
    s_new[1] = '#';
    int j = 2;

    for (int i = 0; i < len; i++)
    {
        s_new[j++] = s[i];
        s_new[j++] = '#';
    }

    s_new[j] = '\0';  // 别忘了哦
    
    return j;  // 返回 s_new 的长度
}

int Manacher()
{
    int len = Init();  // 取得新字符串长度并完成向 s_new 的转换
    int max_len = -1;  // 最长回文长度

    int id;
    int mx = 0;

    for (int i = 1; i < len; i++)
    {
        if (i < mx)
            p[i] = min(p[2 * id - i], mx - i);  // 需搞清楚上面那张图含义, mx 和 2*id-i 的含义
        else
            p[i] = 1;

        while (s_new[i - p[i]] == s_new[i + p[i]])  // 不需边界判断，因为左有'$',右有'\0'
            p[i]++;

        // 我们每走一步 i，都要和 mx 比较，我们希望 mx 尽可能的远，这样才能更有机会执行 if (i < mx)这句代码，从而提高效率
        if (mx < i + p[i])
        {
            id = i;
            mx = i + p[i];
        }

        max_len = max(max_len, p[i] - 1);
    }

    return max_len;
}

例题：

nefu1509

详解见：http://acm.nefu.edu.cn/JudgeOnline/forumDiscuss.php?problem_id=1509

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define ll long long
const int M=5e5+10;
char s[M],s_new[2*M];
long long p[M*2];
long long trie[M*2];
void update(int x)
{
    for(int i=x;i<=M;i+=i&(-i))
    {
        trie[i]+=1;
    }
}
ll query(int x)
{
    ll ans=0;
    for(int i=x;i;i-=i&(-i))
    {
        ans+=trie[i];
    }
    return ans;
}
int init()
{
    s_new[0]='$';
    s_new[1]='#';
    int j=2;
    for(int i=0;s[i];i++)
    {
        s_new[j++]=s[i];
        s_new[j++]='#';
    }
    s_new[j]='\0';
    return j;
}
void mancher()
{
    int len=init();
    int id;
    ll mx=0;
    for(int i=1;i<=len;i++)
    {
        if(i<mx)
        {
            p[i]=min(p[2*id-i],mx-i);
        }
        else
            p[i]=1;
        while(s_new[i-p[i]]==s_new[i+p[i]])
            p[i]++;
        if(mx<i+p[i])
        {
            id=i;
            mx=i+p[i];
        }
    }
}
vector<int>e[M];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
    ll ans=0;
    memset(trie,0,sizeof(trie));
    scanf("%s",s);
    int len=strlen(s);
    mancher();
    for(int i=1;i<=len;i++) e[i].clear();
    for(int i=1;i<=len;i++)
    {
        int pos=(p[2*i]-1)/2;
        e[i-pos].push_back(i);
    }
    for(int i=1;i<=len;i++)
    {
        for(int j=0;j<e[i].size();j++)
        {
            update(e[i][j]);
        }
        ans+=query(i+(p[2*i]-1)/2)-query(i);
    }
    printf("%lld\n",ans);
    }
    return 0;
}