题目链接: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 ws1
int 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;
}