【Codeforces Round #166 (Div. 2)】D. Good Substrings【哈希】

【Codeforces Round #166 (Div. 2)】D. Good Substrings【哈希】
【Codeforces Round #166 (Div. 2)】D. Good Substrings【哈希】


解题思路

以b的个数为关键字,做一个前缀和,然后枚举区间左右端点,i,j,a[j]-a[i-1]如果<=k,说明这段字符是合法的,但是考虑到要求重,要先做一个哈希预处理

base[0]=1ll;
	for(int i=1;i<=len;i++)
	{
		h[i]=(h[i-1]*p*1ll+(s[i]-'a'+1ll));
		base[i]=(base[i-1]*p*1ll);
	}

然后当前字符串的哈希值就是 x = h [ j ] − h [ i − 1 ] ∗ b a s e [ j − i + 1 ] ; x=h[j]-h[i-1]*base[j-i+1]; x=h[j]−h[i−1]∗base[j−i+1];将所有的合法的哈希值存入数组,最后排序,然后 u n i q u e unique unique去重


代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
#define ll long long
#define db double
using namespace std;

const ll p=131;
char c[30];
string s;
ll len,k,cnt,ans;
ll a[1510],base[1510],h[1510],w[20000000];

int main(){
	cin>>s;
	for(int i=0;i<26;i++)
		cin>>c[i];
	scanf("%lld",&k);
	len=s.size();
	s=" "+s;
	for(int i=1;i<=len;i++)
	{
		if(c[s[i]-'a']=='0')
			a[i]=a[i-1]+1;
		else a[i]=a[i-1];
	}
	base[0]=1ll;
	for(int i=1;i<=len;i++)
	{
		h[i]=(h[i-1]*p*1ll+(s[i]-'a'+1ll));
		base[i]=(base[i-1]*p*1ll);
	}
	for(int i=1;i<=len;i++)
	{
		for(int j=i;j<=len;j++)
		{
			if(a[j]-a[i-1]<=k)
			{
				ll x=h[j]-h[i-1]*base[j-i+1];
				w[++cnt]=x;
			}else break;
		}
	}
	sort(w+1,w+cnt+1);
	ll ans=unique(w+1,w+cnt+1)-w-1;
	printf("%lld",ans);
}
上一篇:我想你需要看一下 前端开发常见的几种设计模式


下一篇:awful, awesome, terrible, terrific