「Madoka,不要相信 QB!」伴随着 Homura 的失望地喊叫,Madoka 与 QB 签订了契约。
这是 Modoka 的一个噩梦,也同时是上个轮回中所发生的事。为了使这一次 Madoka 不再与 QB 签订契约,Homura 决定在刚到学校的第一天就解决 QB。然而,QB 也是有许多替身的(但在第八话中的剧情显示它也有可能是无限重生的),不过,意志坚定的 Homura 是不会放弃的——她决定消灭所有可能是 QB 的东西。现在,她已感受到附近的状态,并且把它转化为一个长度为n的字符串交给了学 OI 的你。
现在你从她的话中知道,所有形似于 A + B + A 的字串都是 QB 或它的替身,且|A|>=k,|B|>=1, (位置不同其他性质相同的子串算不同子串,位置相同但拆分不同的子串算同一子串),然后你必须尽快告诉 Homura 这个答案——QB 以及它的替身的数量。
注:对于一个字符串S,|S|表示S的长度。
输入格式
第一行一个字符串S,第二行一个数k。
输出格式
仅一行一个数ans,表示 QB 以及它的替身的数量。
数据范围与提示
对于全部数据,1<=|S|<=1.5*10^4,1<=k<=100,且字符集为所有小写字母。
AC代码如下:
#include<cstring>
#include<cstdio>
using namespace std;
const int N=2e4+110;
int n,k,ans;
int next[N]; char s[N];
int main()
{
scanf("%s%d",s+1,&k),n=strlen(s+1);
for(int t=1,i,j;t <= n-(k<<1);++t)
{
for(i = 1;i <= t;++i)
next[i]=t-1;
for(i = t+1,j = t-1;i <= n;++i)
{
while(j^t-1 && s[i] != s[j+1])
j=next[j];
if(s[i] == s[j+1]) ++j; next[i]=j;
}
for(i = t+1,j = t-1;i <= n;++i)
{
while(j^t-1 && s[i]!=s[j+1])
j=next[j];
if(s[i] == s[j+1]) ++j;
while(j-t+1 >= i-j)
j = next[j];
if(j-t+1 >= k) ++ans;
}
}
printf("%d\n",ans);
return 0;
}
bye~