hdu 6194 后缀数组+RMQ+lcp

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6194

题意

       给你一个字符串,要你求出这个字符串中正好出现k次的子串的个数有多少个。

 

       花了很长一段时间,因为发现板子好像错了,找了好久的bug,就当现在又更新了一遍板子...

做法:

        我们要找连续出现k次的,肯定是从sa数组下手,因为sa[i]数组记录的是字典序为i的串的下标,会连续出现k次的字符串在sa中肯定会是一段长度为k的区间中有公共的前缀,这里用到的lcp就能在O(1)的时间内求区间内最长公共前缀的长度,如果在i~k+i-1的区间内发现最长公共长度为L,那么我们就有L个不同的子串出现了至少k次(以同一个开头开始)

        因为题目要求是正好k次,所以我们要减去至少出现了k+1次的子串的个数,那么我们就能用当前的区间内的l-1,和r+1下手,也就是,但是很明显,会有一些部分被多减了,这个时候我们就要再加上lcp(i-1,i+k)就好啦。

 

 

#include
typedef long long ll;
using namespace std;
const int maxn=100010;
//以下为倍增算法求后缀数组
int wa[maxn],wb[maxn],wv[maxn],Ws[maxn];
int t1[maxn],t2[maxn],c[maxn];
int sa[maxn],Rank[maxn],height[maxn];
int cmp(int *r,int a,int b,int l)
{return r[a]==r[b]&&r[a+l]==r[b+l];}
/**< 传入参数:str,sa,len+1,ASCII_MAX+1 */
void da( int r[],int n,int m)
{
      int i,j,p,*x=wa,*y=wb,*t;
      for(i=0; i=0; i--) sa[--Ws[x[i]]]=i;
      for(j=1,p=1; p=j) y[p++]=sa[i]-j;
            for(i=0; i=0; i--) sa[--Ws[wv[i]]]=y[i];
            for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1; ir) swap(l, r);
    ++l;
    int k=0,len=r-l+1;
    while((1<<(k+1))<=len) ++k;
    return min(dp[l][k], dp[r-(1<>t;
    while(t--){
        scanf("%d%s",&k,str);
        n=strlen(str);
        for(int i=0;i0) ans-=lcp(i-1,i+k-1);
            if(i+k<=n) ans-=lcp(i,i+k);
            if(i-1>0 && i+k<=n) ans+=lcp(i-1,i+k);
        }
        printf("%lld\n",ans);
    }
      return 0;
}

 

你可能感兴趣的:(模板,字符串算法)