HDU 6194 string string string

给出一个字符串求其出现恰好k次的子串数量

对串建立后缀自动机，所有right值为k的节点的value值的和就是答案.

可以在建自动机的时候把 每个前缀加一，然后拓扑更新到pre上，就可以统计每个节点出现的次数。然后val[u]-val[pre[u]]就是u节点子串个数。

#include <iostream>#include <cstdio>#include <string>#include <cstring>using namespace std;typedef long long LL;const int maxn = 2e5 + 10;int sum[maxn],temp[maxn];string s;struct SAM{     int ch[maxn][26], pre[maxn], val[maxn],cnt[maxn];    int last, tot;    void init(){        last = tot = 0;        memset(ch[0], -1, sizeof ch[0]);        pre[0] = -1; val[0] = 0;    }  void extend(int c){        int p = last, np = ++tot;        val[np] = val[p] + 1;        memset(ch[np], -1, sizeof ch[np]);        while(~p && ch[p][c] == -1) ch[p][c] = np, p = pre[p];        if(p == -1) pre[np] = 0;        else{            int q = ch[p][c];            if(val[q] != val[p] + 1){                int nq = ++tot;                memcpy(ch[nq], ch[q], sizeof ch[q]);                val[nq] = val[p] + 1;                pre[nq] = pre[q];                pre[q] = pre[np] = nq;                while(~p && ch[p][c] == q) ch[p][c] = nq, p = pre[p];            }            else pre[np] = q;        }        last = np;        cnt[last]++;    }    void build(const string &s)    {        init();        int t=s.size()*2+10;        for(int i=0;i<t;i++) cnt[i]=0;        for(int i=0;i<s.size();i++) extend(s[i]-'a');    }      void Tsort(const string &s,int k){        int t= s.size()*2+10;        for(int i=0;i<t;i++) sum[i]=0;        for (int i=0;i<=tot;i++) sum[val[i]]++ ;        for (int i=1;i<=s.size();i++) sum[i]+=sum[i-1];        for (int i=0;i<=tot;i++) temp[sum[val[i]]--]=i;    //   for (int i=1;i<=tot+1;i++) printf("%d %d\n",temp[i],val[temp[i]]);        ll ans=0;        for(int i=tot+1;i>=2;i--)        {            int u=temp[i];            if(pre[u]>0) cnt[pre[u]]+=cnt[u];            if(cnt[u]==k)            {                ans+=val[u]-val[pre[u]];            }        }        printf("%lld\n",ans);    }     } sam;int main(){    ios::sync_with_stdio(false); cin.tie(0);    int T,k;    cin>>T;    while(T--)    {        cin>>k>>s;        sam.build(s);     //   sam.debug();        sam.Tsort(s,k);    }    return 0;}/* void debug()    {        for(int i = 0; i <= tot; ++i)        {            printf("id=%d, fa=%d, step=%d,son=[ ", i, pre[i], val[i]);            for(int j = 0; j < 26; ++j)            {                if(ch[i][j]!=-1)                printf("%c,%d ", j+'a',ch[i][j]);            }            printf("]");            printf(" cnt: %d\n",cnt[i]);        }    }*/