Description
“Madoka,不要相信 QB!”伴随着 Homura 的失望地喊叫,Madoka 与 QB 签订了契约.
这是 Modoka 的一个噩梦,也同时是上个轮回中所发生的事.为了使这一次 Madoka 不再与 QB签订契约,Homura 决定在刚到学校的第一天就解决 QB.然而,QB 也是有许多替身的(但在第八话中的剧情显示它也有可能是无限重生的),不过,意志坚定的 Homura 是不会放弃的——她决定
消灭所有可能是 QB 的东西.现在,她已感受到附近的状态,并且把它转化为一个长度为 n 的字符串交给了学 OI 的你.
现在你从她的话中知道 , 所有形似于 A+B+A 的字串都是 QB 或它的替身 , 且len(A)>=k,len(B)>=1 (位置不同其他性质相同的子串算不同子串,位置相同但拆分不同的子串算同一子串),然后你必须尽快告诉 Homura 这个答案——QB 以及它的替身的数量.
Input
第一行一个字符串,第二行一个数 k
Output
仅一行一个数 ans,表示 QB 以及它的替身的数量
Sample Input
【样例输入 1】
aaaaa
1
【样例输入 2】
abcabcabc
2
aaaaa
1
【样例输入 2】
abcabcabc
2
Sample Output
【样例输出 1】
6
【样例输出 2】
8
6
【样例输出 2】
8
HINT
对于 100%的数据:n<=15000 , k<=100,且字符集为所有小写字母
Source
首先KMP是前缀等于后缀,所以它并不能很好的解决子串问题;
但是这个题的正解是n^2的,所以我们可以枚举起点位置i,然后对以起点位置开始到结尾的串重构一个nxt数组,
然后对于起点i,到终点j,就可以通过kmp前缀等于后缀的方法check是否合法了,然后就是跟动物园类似的做法,然后判断一下g>=k即可;
然而我这个题目因为没有想到重构nxt做了好久。。。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #define RG register using namespace std; const int N=15050; int nxt[N],cnt[N],len,k,ans; char s[N],a[N]; void work(int x){ int len2=0; memset(nxt,0,sizeof(nxt)); for(int i=x;i<=len;i++) a[++len2]=s[i]; int j=0; for(int i=2;i<=len2;i++){ while(j&&a[j+1]!=a[i]) j=nxt[j]; if(a[j+1]==a[i]) j++; nxt[i]=j; } j=0; for(int i=2;i<=len2;i++){ while(j&&a[j+1]!=a[i]) j=nxt[j]; if(a[j+1]==a[i]) j++; while(2*j+1>i&&j) j=nxt[j]; if(j>=k) ans++; } } int main(){ scanf("%s",s+1);len=strlen(s+1);scanf("%d",&k); for(int i=1;i<=len;i++) work(i); printf("%d",ans); }