POJ 3415 Common Substrings <后缀数组+单调栈>

题目：传送门。

分析：求长度不小于k的公共子串一共有多少组。直接上代码吧。

代码：

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <string>
using namespace std;

const int MAXN=2e5+4;
typedef long long LL;
int wm[MAXN],wa[MAXN],wb[MAXN],height[MAXN],sa[MAXN];
int *rank;

bool cmp(int *r,int a,int b,int len){
	return r[a]==r[b]&&r[a+len]==r[b+len];
}

void da(int *data,int *sa,int n,int m){
	int *x=wa,*y=wb,*t,i,j,p;
	for(i=0;i<m;++i) wm[i]=0;
	for(i=0;i<n;++i) ++wm[x[i]=data[i]];
	for(i=1;i<m;++i) wm[i]+=wm[i-1];
	for(i=n-1;i>=0;--i) sa[--wm[x[i]]]=i;
	for(j=p=1;p<n;j<<=1,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<m;++i) wm[i]=0;
		for(i=0;i<n;++i) ++wm[x[y[i]]];
		for(i=1;i<m;++i) wm[i]+=wm[i-1];
		for(i=n-1;i>=0;--i) sa[--wm[x[y[i]]]]=y[i];
		for(t=x,x=y,y=t,i=p=1,x[sa[0]]=0;i<n;++i) x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
	}
	rank=x;
}
 
void calHeight(int *data,int *sa,int n){
 	for(int i=0,j,k=0;i<n;height[rank[i++]]=k)
 		for(k?--k:k,j=sa[rank[i]-1];data[i+k]==data[j+k];++k);
}

LL solve(int len,int n,int k){
	int *mark=wm,*s=wb,top=0;
	LL sum[2],res=0;
	for(int i=1;i<=n;++i){
		if(height[i]<k){
			top=sum[0]=sum[1]=0;
		}
		else{
			for(int j=top;j&&s[j]>height[i]-k+1;--j){
				sum[mark[j]]-=(s[j]-height[i]+k-1);
				s[j]=height[i]-k+1;
			}
			s[++top]=height[i]-k+1;
			if(sa[i-1]<len) mark[top]=0;
			if(sa[i-1]>len) mark[top]=1;
			sum[mark[top]]+=height[i]-k+1;
			if(sa[i]>len) res+=sum[0];
			if(sa[i]<len) res+=sum[1]; 
		}
	}
	return res;
}

int data[MAXN];

int main(){
	int k;
 	while(cin>>k,k){
 		string str1,str2;
 		cin>>str1>>str2;
 		int len1=str1.size(),len2=str2.size();
 		for(int i=0;i<len1;++i) data[i]=str1[i];
 		data[len1]='#';
 		for(int i=0;i<len2;++i) data[len1+1+i]=str2[i];
 		data[len1+len2+1]=0;
 		da(data,sa,len1+len2+2,200);
 		calHeight(data,sa,len1+len2+1);
 		cout<<solve(len1,len1+len2+1,k)<<endl;
	 }
	 return 0;
}