hdu6194 string string string

链接

http://acm.hdu.edu.cn/showproblem.php?pid=6194

题解

我竟然质疑自己过了将近20题的模板T_T，而且最后发现模板对的不得了，是我自己想错了
统计出现次数大于等于$k$的很简单吧，就在后缀数组中求以每个后缀为结尾的长度为$k$的区间的串的长度，再减去重复算的
这题就用出现次数$\ge k$的减去出现次数$\ge k+1$的就好了

代码

#include <bits/stdc++.h>
#define maxn 100010
#define maxk 17
#define cl(x) memset(x,0,sizeof(x))
using namespace std;
typedef long long ll;
struct SuffixArray
{
    int sa[maxn], rank[maxn], ws[maxn], wv[maxn], wa[maxn], wb[maxn], height[maxn], st[maxk+2][maxn], N;
    bool cmp(int *r, int a, int b, int l){return r[a]==r[b] and r[a+l]==r[b+l];}
    void clear()
    {
        cl(sa), cl(rank), cl(ws), cl(wv), cl(wa), cl(wb), cl(height);
    }
    void build(int *r, int n, int m)
    {
        N=n;
        n++;
        int i, j, k=0, p, *x=wa, *y=wb, *t;
        for(i=0;i<m;i++)ws[i]=0;
        for(i=0;i<n;i++)ws[x[i]=r[i]]++;
        for(i=1;i<m;i++)ws[i]+=ws[i-1];
        for(i=n-1;i>=0;i--)sa[--ws[x[i]]]=i;
        for(p=j=1;p<n;j<<=1,m=p)
        {
            for(p=0,i=n-j;i<n;i++)y[p++]=i;
            for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
            for(i=0;i<n;i++)wv[i]=x[y[i]];
            for(i=0;i<m;i++)ws[i]=0;
            for(i=0;i<n;i++)ws[wv[i]]++;
            for(i=1;i<m;i++)ws[i]+=ws[i-1];
            for(i=n-1;i>=0;i--)sa[--ws[wv[i]]]=y[i];
            for(t=x,x=y,y=t,p=1,i=1,x[sa[0]]=0;i<n;i++)
                x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
        }

        for(i=0;i<n;i++)rank[sa[i]]=i;

        for(i=0;i<n-1;height[rank[i++]]=k)
            for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);

    }
    void build_st()
    {
        int i, k;
        for(i=1;i<=N;i++)st[0][i]=height[i];
        for(k=1;k<=maxk;k++)
            for(i=1;i+(1<<k)-1<=N;i++)
                st[k][i]=min(st[k-1][i],st[k-1][i+(1<<k-1)]);
    }
    int lcp(int x, int y)
    {
        int l=rank[x], r=rank[y];
        if(l>r)swap(l,r);
        if(l==r)return N-sa[l];
        int t=log2(r-l);
        return min(st[t][l+1],st[t][r-(1<<t)+1]);
    }
}SA;
int r[maxn], n, k;
char s[maxn];
ll ans, T;
int main()
{
    ll i;
    scanf("%lld",&T);
    while(T--)
    {
        scanf("%d%s",&k,s);
        n=strlen(s);
        for(i=0;i<n;i++)r[i]=s[i]; r[n]=0;
        SA.clear();
        SA.build(r,n,300);
        SA.build_st();
        ans=0;
        for(i=k;i<=n;i++)
        {
            auto x=SA.lcp(SA.sa[i],SA.sa[i-k+1]), y=SA.lcp(SA.sa[i],SA.sa[i-k]);
            ans+=x-y;
        }
        k++;
        for(i=k;i<=n;i++)
        {
            auto x=SA.lcp(SA.sa[i],SA.sa[i-k+1]), y=SA.lcp(SA.sa[i],SA.sa[i-k]);
            ans-=x-y;
        }
        printf("%lld\n",ans);
    }
    return 0;
}