Mail.Ru Cup 2018 Round 3 E. Check Transcription（字符串哈希）

题目链接：http://codeforces.com/contest/1056/problem/E

题意

        给你一个01串s和一个字符串t，你可以让0等于串r0，1等于r1，要求r0不等于r1且都非空，将所有的0改变成r0，所有的1改变成r1后要等于字符串t。求有多少组r0，r1满足上述条件。

思路

        枚举0对应的字符串的长度len0，则可以根据该长度计算出1对应的字符串的长度len1。假设0出现的第一个位置为p0，1出现的第一个位置为p1，则r0为字符串t在[p0, p0+len0-1]的取值，r1为字符串t在[p1, p1+len1-1]的取值。接着可以采用字符串哈希的方式截取字符串t中对应长度的字符串于r0，r1串进行哈希值的比对，如果全部满足要求则ans++。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
const ll mod=1e9+7;
const ll base=233;

int lens,lent;
char s[maxn],t[maxn*10];
ll Hash[maxn*10],p[maxn*10];
//计算hash值
void calcHash()
{
	Hash[0]=t[0];
	p[0]=1;
	for(int i=1;i<lent;i++){
		Hash[i]=(Hash[i-1]*base+t[i])%mod;
		p[i]=(p[i-1]*base)%mod;
	}
	Hash[lent]=(Hash[lent-1]*base)%mod;
}
//获得[l,r]区间上字符串的hash值
ll getHash(int l,int r)
{
	if(l==0) return Hash[r];
	else return (Hash[r]-(Hash[l-1]*p[r-l+1])%mod+mod)%mod;
}

int main()
{
	scanf("%s %s",s,t);
	lens=strlen(s);
	lent=strlen(t);
	calcHash();
	//cnt1为s[0]字符的个数，cnt2为于s[0]不同的字符的个数
	int cnt1=0,cnt2=0,fi=-1;
	for(int i=0;i<lens;i++){
		if(s[i]==s[0]) cnt1++;
		else{
			cnt2++;
			if(fi==-1) fi=i;
		}
	}

	ll ans=0;
	for(int i=1;i<=lent;i++){
		if(lent-cnt1*i<=0) break;
		if((lent-cnt1*i)%cnt2!=0) continue; //若不能整除就continue
		int num=(lent-cnt1*i)/cnt2;

		ll hash1=getHash(0,i-1); //字符串r0的哈希值
		ll hash2=getHash(fi*i,fi*i+num-1); //字符串r1的哈希值
		if(hash1==hash2) continue;

		ll cnt=1,l=0;
		for(int j=0;j<lens;j++){
			if(s[j]==s[0]){
				if(getHash(l,l+i-1)!=hash1){ //将t中[l,l+i-1]区间哈希值与hash1比较
					cnt=0;
					break;
				}
				l+=i;
			}
			else{
				if(getHash(l,l+num-1)!=hash2){ //将t中[l,l+num-1]区间哈希值与hash2比较
					cnt=0;
					break;
				}
				l+=num;
			}
		}
		ans+=cnt;
	}
	printf("%lld\n",ans);
	return 0;
}