HDU 3518 Boring counting(后缀数组 所有不重叠字串出现两次以上的次数)

来源:互联网 发布:金刚经说什么 知乎 编辑:程序博客网 时间:2024/06/03 17:51

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


这个题目又是后缀数组的一个应用,同样和上面几题一样都是在height数组上做文章

这个题目首先要明确一点就是题目要求求全部个数,那么我们就可以按照我们的想法按照字串长度为1到(n+1)/2来枚举

枚举的时候用到一最前面的一个题目的解题思路搞定重复问题,然后题目要求的字串不重叠问题就搞定了

下面来说一下这样搞为什么能同样长度的字串不重复

其实这个也不用证明,因为能想的通,想一下,假设我们现在要求的是长度为K的不同子串出现的次数,那么我们

就按照下面一段代码来执行

 for(int i=1;i<=n;i++)    {        if(height[i]<m)        {            mmax=sa[i];            mmin=sa[i];            flag=true;        }        else        {            mmax=max(mmax,max(sa[i],sa[i-1]));            mmin=min(mmin,min(sa[i],sa[i-1]));            if(flag && mmax-mmin>=m)            {                answer++;                flag=false;            }        }
因为height数组是相邻两个排名的后缀的最长前缀,那么前缀相同的排名一定是相邻了,意思就是前缀相同的一系列的后缀串一定是排名靠着的

所以这个题目还是按照height分组,每一个符合条件的分组一定是一个答案,而且不会是重复的答案,我们用flag标记每一个新的分组,出现一次

height小于m的表示分组从此被打断,重新来一次,表示后缀数组本质上还是在height上做文章,关键还是要理解height数组是什么意思,然后就

主要在height数组上做文章,前面求height数组的过程都是基本上一样的,关键还是对结果的操作!

#include <iostream>#include <stdio.h>#include <algorithm>#include <math.h>#include <string.h>using namespace std;#define maxn 1100#define ws ws1int answer;int wa[maxn],wb[maxn],wv[maxn],ws[maxn];int cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}void da(char *r,int *sa,int n,int m){int i,j,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(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++) 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,x[sa[0]]=0,i=1;i<n;i++)x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;}return;}int rank[maxn],height[maxn];void calheight(char *r,int *sa,int n){int i,j,k=0;for(i=1;i<=n;i++) rank[sa[i]]=i;//在计算height的时候顺便就把rank计算出来了,反正也要用for(i=0;i<n;height[rank[i++]]=k)for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);return;}char rec[maxn];int sa[maxn];int n;int check(int m){    int mmax=0,mmin=n;    bool flag=true;    for(int i=1;i<=n;i++)    {        if(height[i]<m)        {            mmax=sa[i];            mmin=sa[i];            flag=true;        }        else        {            mmax=max(mmax,max(sa[i],sa[i-1]));            mmin=min(mmin,min(sa[i],sa[i-1]));            if(flag && mmax-mmin>=m)            {                answer++;                flag=false;            }        }    }    return 0;}int main(){    int i,j,k;    int left,right;    int mid;    while(scanf("%s",rec))    {        if(strcmp("#",rec)==0)        return 0;        n=strlen(rec);     da(rec,sa,n+1,200);//这里是n+1 因为看这个函数里面是 < n 的     calheight(rec,sa,n);//注意这里面是 n 了因为看函数里面是 <=n 的,所以这里要注意    if(n==1)    {        printf("0\n");        continue;    }     answer=0;     for(i=1;i<n/2+1;i++)     check(i);     printf("%d\n",answer);    }    return 0;}