1. 程式人生 > >BZOJ_P3620 似乎在夢中見過的樣子(KMP)

BZOJ_P3620 似乎在夢中見過的樣子(KMP)

BZOJ傳送門
Time Limit: 15 Sec Memory Limit: 128 MB
Submit: 351 Solved: 194
[Submit][Status][Discuss]
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

Sample Output
【樣例輸出 1】
6

【樣例輸出 2】
8

HINT
對於 100%的資料:n<=15000 , k<=100,且字符集為所有小寫字母

Source
2014湖北省隊互測week2

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std; #define N 15005 char s[N];int f[N]; int k,l,ans,lim; int main(){ scanf("%s%d",s+1,&k);l=strlen(s+1);lim=l-k*2; for(int p=0;p<lim;p++){ for(int j=0,i=2;i+p<l;i++){ while(j&&s[j+p+1]!=s[i+p]) j=f[j]; if(s[i+p]==s[j+p+1]) j++;f[i]=j; } for
(int j=0,i=k+1;i+p<=l;i++){ while(j&&s[i+p]!=s[j+p+1]) j=f[j]; if(s[i+p]==s[j+p+1]) j++; while((j<<1)>=i) j=f[j];if(j>=k) ans++; } } printf("%d\n",ans); return 0; }