杭电多校第二场 I Love Palindrome String (Pam+Hash)

给你一个字符串,问长度为0~|s| 满足l~r为回文串 并且l~(l+r)/2 也为回文串的数量

由题意可知如果长度为偶数则 需要满足l~(l+r)/2 == (l+r)/2+1~r

如果长度为奇数则需要满足l~(l+r)/2==(l+r)/2~r 

对于回文树上的每个节点,我们预处理每个节点的回文串的末尾字符所在的下标 然后hash特判回文串是否满足

然后根据长度累计答案

#include//空间O(N*字符集大小) 时间O(N*log(字符集大小))
#define ll long long
#define ull unsigned long long
#define maxn 300010   //节点个数
using namespace std;
const ull base=201326611;
char a[maxn];
ll ans[maxn];
ull Hash[maxn];
ull p[maxn];
ull get_hash(int l, int r)
{
    if(l==0)return Hash[r];
    return Hash[r]-Hash[l-1]*p[r-l+1];
}
bool check(int l, int r)
{
    int le=r-l+1;
    int mid=(r+l)>>1;
    if(le&1)
    {
        return get_hash(l,mid)==get_hash(mid,r);
    }
    else return get_hash(l,mid)==get_hash(mid+1,r);
}
struct pam
{
    int len[maxn];  //len[i] 节点i的回文串长度
    int nxt[maxn][26]; //nxt[i]['c'] 节点i的回文串在两边添加字符c后变成的回文串编号
    int fail[maxn]; //fail[i] 指向i的最长回文后缀所在的节点 且不为i
    int cnt[maxn]; //cnt[i] 节点i表示的回文串在S中出现的次数
    int s[maxn]; //s[i] 第i次添加的字符
    int num[maxn];
    int id[maxn];
    int last; //指向以字符串中上一个位置结尾的回文串的节点
    int cur; //指向由next构成的树中当前回文串的父亲节点(即当前回文串是cur左右两边各拓展一个字符得来)
    int p; // 添加的节点个数
    int n; // 添加的字符串个数
    int newnode(int l)  //新建节点
    {
        for(int i=0;i<=25;i++)nxt[p][i]=0; // 消除子节点
        cnt[p]=num[p]=0;  //节点p为新回文串所以出现次数为0
        len[p]=l;
        return p++;
    }
    inline void init()
    {
        p=n=last=0;
        newnode(0);  //偶节点
        newnode(-1); //奇节点
        s[0]=-1;
        fail[0]=1;
    }
    int get_fail(int x) //找到可以插入的节点
    {
        while(s[n-1-len[x]]!=s[n])x=fail[x];
        return x;
    }
    inline void add(int c)
    {
        c-='a';
        s[++n]=c;
        int cur=get_fail(last); // 找到可以插入的节点并当做父节点
        if(!nxt[cur][c])
        {
            int now=newnode(len[cur]+2);
            fail[now]=nxt[get_fail(fail[cur])][c]; //从父节点的回文后缀开始找,找到一个s[l-1]=s[n]的点则出边的点为最长回文后缀
            nxt[cur][c]=now;
            num[now]=num[fail[now]]+1;
        }
        last=nxt[cur][c];  //成为新的上一个位置
        id[last]=n;
        cnt[last]++;
    }
    void Count() //统计本质相同的回文串的出现次数。与位置无关
    {
        for(int i=p-1;i>=0;i--)
        {
            cnt[fail[i]]+=cnt[i]; //每个节点会比父节点先算完,于是父节点能加到所有的子节点
        }
        for(int i=2;i

 

你可能感兴趣的:(回文树学习)