題目連結: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;}