给一个原串,以及那些字符是坏的,现在问你可以从原串中取出多少个不同子串,使得其所含的坏字符的个数不超过一个定数。
这个题目网上有各种各样的解法。如hash,tire。
我说一下我的解法。
解法一:后缀自动机dp。f[][]保存到达某个状态,前面已经有的坏字符的个数的时候的字符串数量。这样按照拓扑序列一直递推下去就可以了。时间复杂度为O(N2)。
解法二:后缀自动机,预处理。对于字符串,每个位置保存它可以最前走到那个位置。然后,在自动机上增加信息,保存位置。然后直接按照距离判断就可以了。对于每个节点,直接缩小范围,更新答案。时间复杂度为O(N)。
召唤代码君:
#include <iostream> #include <cstdio> #include <cstring> #define maxn 3555 using namespace std; char real[26],s[maxn]; int next[maxn][26],pre[maxn],step[maxn]; int N,last,n,L; int p,q,np,nq,ans=0; int f[maxn],tag[maxn],sum[maxn]; void insert(int x,int ggg) { p=last,np=++N,step[np]=step[p]+1,last=np,tag[np]=ggg; for (; p!=-1 && next[p][x]==0; p=pre[p]) next[p][x]=np; if (p==-1) return; q=next[p][x]; if (step[q]==step[p]+1) { pre[np]=q; return ; } nq=++N,step[nq]=step[p]+1,pre[nq]=pre[q],tag[nq]=ggg; for (int i=0; i<26; i++) next[nq][i]=next[q][i]; pre[np]=pre[q]=nq; for (; p!=-1 && next[p][x]==q; p=pre[p]) next[p][x]=nq; } int main() { scanf("%s",s+1); scanf("%s",real); scanf("%d",&n); pre[0]=-1; L=strlen(s+1); for (int i=1; s[i]; i++) insert(s[i]-‘a‘,i); sum[0]=0; for (int i=1; i<=L; i++) { sum[i]=sum[i-1]; if (real[s[i]-‘a‘]==‘0‘) sum[i]++; } f[L+1]=L+1; for (int i=L; i>=1; i--) { int k=min(f[i+1],i+1); for ( ;k>1 && sum[i]-sum[k-2]<=n; k--) ; f[i]=k; } for (int i=1; i<=N; i++) { int l=tag[i]-step[i]+1,r=tag[i]-step[pre[i]]; if (f[tag[i]]>l) l=f[tag[i]]; if (l<=r) ans+=r-l+1; } printf("%d\n",ans); return 0; }
CF271D_Good Substrings
时间: 2024-10-05 12:43:32