解题思路
以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);
}