题目描述
给出一个字符串S,求出其子串中满足ABA形式且|A|≥k,|B|≥1的数目。
思路
首先明确这道题n2暴力是可以过的,而题目中定义可以转化为A为子串的公共前后缀且长度大于k,小于整个子串长度的1/2,所以我们考虑用KMP来求公共前后缀,然后暴力判断即可。不过考虑到每次判断时要不断往前跳,有超时的可能性,我们考虑优化。首先我们枚举每个字符串的左端点,显然枚举到n-k*2就可以了。接下来对于l~n这一段进行一次KMP。对于S[l...r]这一段,我们考虑处理出来的p[i]=j,那么如果j<l+k-1,那么最长公共前后缀不满足条件,g[i]=i,也就是这一段只能全部作为A字符串;否则满足条件,就以g[j]为条件,因为我们要使它尽可能短,这样对后续状态影响小。最后只要判断记录的答案位置是否这一段的中间位置即可。
代码
#include <bits/stdc++.h> using namespace std; const int MAXN=2e4+10; char s[MAXN]; int pre[MAXN],n,g[MAXN]; int main() { int k; scanf(" %s",s+1); scanf("%d",&k); n=strlen(s+1); int ans=0; for(int l=1;l<=n-(k<<1);l++) { pre[l]=l-1; int j=l-1; g[l]=l; for(int p=l;p<n;p++) { while(j>l-1&&s[p+1]!=s[j+1])j=pre[j]; if(s[p+1]==s[j+1])j++; pre[p+1]=j; g[p+1]=(j<l+k-1)?p+1:g[j]; if(g[p+1]<(l+p+1)/2) ans++; } } printf("%d",ans); return 0; }
原文地址:https://www.cnblogs.com/fangbozhen/p/11623385.html
时间: 2024-10-01 07:03:41