题意:告诉你一个字符串和k , 求这个字符串中有多少不同的子串恰好出现了k 次。
链接:HDU 6194
思路:后缀数组搞起!
首先处理出来Height数组,然后初始化ST表,那么我们从头开始RMQ查询长度为k的区间,那么区间最小值那个前缀至少出现了k次,然后我们将区间左右扩展,目的就是为了去除超过k次的,然后减掉多出来的就可以。
#include<bits/stdc++.h>
using namespace std;
#define N 500000
int st[N][50],n,rk[N],sa[N],height[N],tmp[N],cnt[N];
char s[N];
void suffixarray(int n,int m) { //n为字符串长度,m为基数排序的阙值
int i,j,k;n++;
for(i=0;i<n*2+5;i++) rk[i]=sa[i]=height[i]=tmp[i]=0;
for(i=0;i<m;i++) cnt[i]=0;
for(i=0;i<n;i++) cnt[rk[i]=s[i]]++;
for(i=1;i<m;i++) cnt[i]+=cnt[i-1];
for(i=0;i<n;i++) sa[--cnt[rk[i]]]=i;
// for(i=0;i<n;i++)
// cout<<"sa="<<sa[i]<<' ';
// cout<<endl;
for(k=1;k<=n;k<<=1)
{
for(i=0;i<n;i++)
{
j=sa[i]-k;
if(j<0) j+=n;
tmp[cnt[rk[j]]++]=j;
}
sa[tmp[cnt[0]=0]]=j=0;
for(i=1;i<n;i++)
{
if(rk[tmp[i]]!=rk[tmp[i-1]]||rk[tmp[i]+k]!=rk[tmp[i-1]+k]) cnt[++j]=i;
sa[tmp[i]]=j;
}
memcpy(rk,sa,n*sizeof(int));
memcpy(sa,tmp,n*sizeof(int));
if(j>=n-1) break;
}
for(j=rk[height[i=k=0]=0];i<n-1;i++,k++)
{
while(~k&&s[i]!=s[sa[j-1]+k]) height[j]=k--,j=rk[sa[j]+1];
}
}
void RMQ_init()
{
// cout<<"n="<<n<<endl;
int i;
for(i=0;i<=n;i++)
st[i][0]=height[i];
for(i=1;(1<<i)<=n;i++)
{
// cout<<i<<endl;
for(int j=2;j+(1<<i)-1<=n;j++)
{
st[j][i]=min(st[j][i-1],st[j+(1<<(i-1))][i-1]);
}
}
// for(int i=1;i<=n;i++)
// cout<<height[i]<<' '<<s+sa[i]<<endl;
// for(int i=2;i<=n;i++)
// {
// for(int j=0;j<=5;j++)
// cout<<st[i][j]<<' ';
// cout<<endl;
// }
}
int RMQ(int l,int r)
{
if(l==r) return strlen(s+sa[l]);
l++;
r++;
int len=log2(r-l);
// cout<<len<<endl;
return min(st[l][len],st[r-(1<<len)][len]);
}
void init()
{
n=strlen(s);
suffixarray(n,500);
RMQ_init();
}
int main()
{
//test
int t;
cin>>t;
while(t--)
{
int k;
scanf("%d",&k);
scanf("%s",s);
init();
int ans=0;
for(int i=1;i+k-1<=n;i++)
{
int l=i,r=i+k-1;
int len=RMQ(l,r);
int more=max((l>1?RMQ(l-1,r):0),(r<n?RMQ(l,r+1):0));
//cout<<"len="<<len<<" more="<<more<<endl;
ans+=max(0,(len-more));
}
cout<<ans<<endl;
}
}