POJ-3415 Common Substrings 后缀数组

　　题目链接：http://poj.org/problem?id=3415

　　求两个串的长度不小于K的公共字串的个数。

　　利用height[]来维护一个单调递增的栈，即栈保存的是满足要求的heigiht数组的最小值，附加维护栈中相邻两个元素之间的个数。

  1 //STATUS:C++_AC_813MS_6532KB

  2 #include<stdio.h>

  3 #include<stdlib.h>

  4 #include<string.h>

  5 #include<math.h>

  6 #include<iostream>

  7 #include<string>

  8 #include<algorithm>

  9 #include<vector>

 10 #include<queue>

 11 #include<stack>

 12 #include<map>

 13 #include<set>

 14 using namespace std;

 15 //define

 16 #define pii pair<int,int>

 17 #define mem(a,b) memset(a,b,sizeof(a))

 18 #define lson l,mid,rt<<1

 19 #define rson mid+1,r,rt<<1|1

 20 #define PI acos(-1.0)

 21 //typedef

 22 typedef __int64 LL;

 23 typedef unsigned __int64 ULL;

 24 //const

 25 const int N=200010;

 26 const int INF=0x3f3f3f3f;

 27 const int MOD=1000007,STA=8000010;

 28 const LL LNF=1LL<<60;

 29 const double EPS=1e-8;

 30 const double OO=1e15;

 31 //Daily Use ...

 32 template<class T> inline T Min(T a,T b){return a<b?a:b;}

 33 template<class T> inline T Max(T a,T b){return a>b?a:b;}

 34 template<class T> inline T Min(T a,T b,T c){return min(min(a, b),c);}

 35 template<class T> inline T Max(T a,T b,T c){return max(max(a, b),c);}

 36 //

 37 

 38 struct Node{

 39     int l,cnt;

 40 }sta[N];

 41 

 42 char s[N/2];

 43 int num[N];

 44 int sa[N],t1[N],t2[N],c[N],rank[N],height[N];

 45 int n,m,len1,len2,K;

 46 LL ans;

 47 

 48 void build_sa(int s[],int n,int m)

 49 {

 50     int i,k,p,*x=t1,*y=t2;

 51     //第一轮基数排序

 52     for(i=0;i<m;i++)c[i]=0;

 53     for(i=0;i<n;i++)c[x[i]=s[i]]++;

 54     for(i=1;i<m;i++)c[i]+=c[i-1];

 55     for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;

 56     for(k=1;k<=n;k<<=1){

 57         p=0;

 58         //直接利用sa数组排序第二关键字

 59         for(i=n-k;i<n;i++)y[p++]=i;

 60         for(i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;

 61         //基数排序第一关键字

 62         for(i=0;i<m;i++)c[i]=0;

 63         for(i=0;i<n;i++)c[x[y[i]]]++;

 64         for(i=1;i<m;i++)c[i]+=c[i-1];

 65         for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];

 66         //根据sa和x数组计算新的x数组

 67         swap(x,y);

 68         p=1;x[sa[0]]=0;

 69         for(i=1;i<n;i++)

 70             x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;

 71         if(p>=n)break;   //已经排好序，直接退出

 72         m=p;     //下次基数排序的最大值

 73     }

 74 }

 75 

 76 void getHeight(int s[],int n)

 77 {

 78     int i,j,k=0;

 79     for(i=0;i<=n;i++)rank[sa[i]]=i;

 80     for(i=0;i<n;i++){

 81         if(k)k--;

 82         j=sa[rank[i]-1];

 83         while(s[i+k]==s[j+k])k++;

 84         height[rank[i]]=k;

 85     }

 86 }

 87 

 88 void slove(int flag)

 89 {

 90     int i,j,top=-1;

 91     LL sum=0;

 92     for(i=2;i<=n;i++){

 93         if(height[i]<K){top=-1;sum=0;continue;}

 94         j=0;

 95         if((sa[i-1]<len1)==flag)j=1,sum+=(LL)height[i]-K+1;

 96         for(;top>=0 && sta[top].l>=height[i];top--){

 97             sum-=(sta[top].l-height[i])*(LL)sta[top].cnt;

 98             j+=sta[top].cnt;

 99         }

100         sta[++top].l=height[i];

101         sta[top].cnt=j;

102         if((sa[i]<len1)!=flag)

103             ans+=sum;

104     }

105 }

106 

107 int main()

108 {

109  //   freopen("in.txt","r",stdin);

110     int i,j;

111     while(~scanf("%d",&K) && K)

112     {

113         scanf("%s",s);

114         len1=strlen(s);

115         for(i=0;i<len1;i++)num[i]=s[i]-'A'+1;

116         scanf("%s",s);

117         len2=strlen(s);

118         num[i++]=80;

119         for(j=0;j<len2;i++,j++)num[i]=s[j]-'A'+1;

120         num[n=len1+len2+1]=0;

121         m=81;

122 

123         build_sa(num,n+1,m);

124         getHeight(num,n);

125         ans=0;

126         slove(1);

127         slove(0);

128 

129         printf("%I64d\n",ans);

130     }

131     return 0;

132 }