回文树

来源:互联网 发布:淘宝店铺优惠券显示 编辑:程序博客网 时间:2024/06/07 18:21

参考代码:http://blog.csdn.net/gatevin/article/details/44783041
首先要知道,一个长度为n的字符串最多有n个本质不同的回文串
所以回文树中最多有n个节点。
回文树:把本质不同的回文串用fail和nxt数组联系起来

const int maxn = 300000+10;const int N = 26;struct PT{    int nxt[maxn][N],fail[maxn];    int p,last,cnt[maxn],len[maxn];//p:添加的节点个数 n:添加的字符个数 last:以新添加的字符结尾的最长字符串在树中节点编号    int n,S[maxn];//S]i]:第i个字符 cnt[i]:第i个节点的字符串个数    int newnode(int l){        for(int i=0;i<N;i++) nxt[p][i]=0;        cnt[p]=0; len[p]=l;        return p++;    }    void init(){        last=p=n=0;        S[0]=-1,fail[0]=1;//fail[0]赋值为1,不能赋值为0        newnode(0),newnode(-1);//先建立0节点,再建立-1节点!!!    }    int get_fail(int x){        while(S[n]!=S[n-len[x]-1]) x=fail[x];        return x;    }    void add(int c){        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];//求now节点的最长回文后缀 cur是它除去两侧‘c’分符后的字符串编号            nxt[cur][c]=now;        }        last=nxt[cur][c];        cnt[last]++;    }    void count(){        for(int i=p-1;i>=0;i--) cnt[fail[i]]+=cnt[i];    }};
0 0