利用hash求最长公共子串的长度

#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
const ll mod=1000000007;
const ll p=10000019;
const ll maxn=1010;
ll powp[maxn],h1[maxn]={0},h2[maxn]={0};
vector<pair<int,int> > pr1,pr2;

void init(int len){ //初始化powp 数组 
	powp[0]=1;
	for(int i=1;i<=len;i++){
		powp[i]=(powp[i-1]*p)%mod;
//		printf("%d ",powp[i-1]);
	}
}
 
void calh(ll h[],string &str){ //计算字符串哈希 
	h[0]=str[0];
	for(int i=1;i<str.length();i++){
		h[i]=(h[i-1]*p+str[i])%mod;
	}
}

int callsinglesubh(ll h[],int i,int j){ ///计算h[i...j]
	if(i==0) return h[j];
	return ((h[j]-h[i-1]*powp[j-i+1])%mod+mod)%mod;
}
//遍历计算所有子串的哈希值! 
void calsubh(ll h[],int len,vector<pair<int,int> >&pr){
	for(int i=0;i<len;i++){
		for(int j=i;j<len;j++){
			int hashvalue=callsinglesubh(h,i,j);
			pr.push_back(make_pair(hashvalue,j-i+1));
		}
	}
}

int getmax(){ //计算pr1和pr2中相同的hash值,维护最大长度! 
	int ans=0;
	for(int i=0;i<pr1.size();i++){
		for(int j=0;j<pr2.size();j++){
			if(pr1[i].first == pr2[j].first) //如果子串哈希相等,则子串相等 ! 
			ans=max(ans,pr1[i].second); //看该子串长度是否目前最大! 
		}
	}
	return ans; 
}

int main(){
	string str1,str2;
	getline(cin,str1);
	getline(cin,str2);
	init(max(str1.length(),str2.length()));
	calh(h1,str1);
	calh(h2,str2);
	calsubh(h1,str1.length(),pr1);
	calsubh(h2,str2.length(),pr2);
	printf("ans=%d\n",getmax()); 
//	cout<
	return 0;
} 

你可能感兴趣的:(PAT&PTA)