HDU 6194 string string string

题目链接：string string string
题意：问一个给定字符串中有多少个正好出现k次的子串
题解：考虑$k=1$和$k\neq 1$的情况，$k\neq 1$的时候直接用后缀数组，然后用单调栈维护一个长度为$k-1$窗口的最小值（因为height数组代表的是两个后缀之间的关系），出栈的时候更新答案。$k=1$的时候考虑当前后缀$sa[i]$与$sa[i-1]$，$sa[i+1]$的$lcp$，即$height[i]$和$height[i+1]$中大的那个用长度减去加到答案里面即可。

#include <bits/stdc++.h>

using namespace std;

const int N = 110000;
int r[N],sa[N],rk[N],het[N],wa[N],wb[N],wv[N],wt[N];

inline bool cmp(int* r,int a,int b,int l){
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int* r,int* sa,int n,int m){
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0;i<m;i++)
        wt[i]=0;
    for(i=0;i<n;i++)
        wt[x[i]=r[i]]++;
    for(i=1;i<m;i++)
        wt[i]+=wt[i-1];
    for(i=n-1;i>=0;i--)
        sa[--wt[x[i]]]=i;
    for(j=1,p=1;p<n;j*=2,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++)
            wt[i]=0;
        for(i=0;i<n;i++)
            wt[wv[i]]++;
        for(i=1;i<m;i++)
            wt[i]+=wt[i-1];
        for(i=n-1;i>=0;i--)
            sa[--wt[wv[i]]]=y[i];
        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}
void calheight(int* r,int* sa,int n){
    int i,j,k=0;
    for(i=1;i<=n;i++)
        rk[sa[i]]=i;
    for(i=0;i<n;het[rk[i++]]=k)
        for(k?k--:0,j=sa[rk[i]-1];r[i+k]==r[j+k];k++);
}

char s[N];
int vis[1000];
pair<int,int> q[N];

int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int k,len,x=0;
        scanf("%d %s",&k,s);
        len=strlen(s);
        for(int i=0;i<len;i++)
            r[x++]=s[i]+1;
        r[x]=0;
        memset(sa,0,sizeof sa);
        memset(het,0,sizeof het);
        da(r,sa,x+1,1000);
        calheight(r,sa,x);
        if(k==1){
            long long ans=0;
            for(int i=1;i<=x;i++)
                ans+=max(0,len-sa[i]-max(het[i],het[i+1]));
            printf("%lld\n",ans);
            continue;
        }
        long long ans=0;
        int lf=1,rt=1;
        for(int i=2;i<=x+1;i++){
            int mn=i-1;
            while(lf<rt&&het[i]<q[rt-1].first){
                if(i-q[rt-1].second==k)
                    ans+=q[rt-1].first-(rt-lf>1?max(q[rt-2].first,het[i]):het[i]);
                mn=min(mn,max(1,(het[q[rt-1].second]>=het[i]?q[rt-1].second-1:q[rt-1].second)));
                rt--;
            }
            q[rt++]=make_pair(het[i],mn);
        }
        printf("%lld\n",ans);
    }
    return 0;
}