【bzoj3620】【似乎在夢中見過的樣子】【kmp】
阿新 • • 發佈：2019-01-07
Description
“Madoka,不要相信 QB！”伴隨著 Homura 的失望地喊叫,Madoka 與 QB 簽訂了契約. 這是 Modoka 的一個噩夢,也同時是上個輪迴中所發生的事.為了使這一次 Madoka 不再與 QB簽訂契約,Homura 決定在剛到學校的第一天就解決 QB.然而,QB 也是有許多替身的(但在第八話中的劇情顯示它也有可能是無限重生的),不過,意志堅定的 Homura 是不會放棄的——她決定 消滅所有可能是 QB 的東西.現在，她已感受到附近的狀態,並且把它轉化為一個長度為 n 的字串交給了學 OI 的你. 現在你從她的話中知道 , 所有形似於 A+B+A 的字串都是 QB 或它的替身 , 且Input
第一行一個字串,第二行一個數 kOutput
僅一行一個數 ans,表示 QB 以及它的替身的數量Sample Input
【樣例輸入 1】aaaaa
1
【樣例輸入 2】
abcabcabc
2
Sample Output
【樣例輸出 1】6
【樣例輸出 2】
8
HINT
對於 100%的資料：n<=15000 , k<=100，且字符集為所有小寫字母
題解：列舉左端點,然後擴充套件kmp即可.
程式碼：
#include<iostream> #include<cstdio> #include<cstring> #define N 20000 using namespace std; char s[N],ss[N]; int ans,k,j,fail[N],len; int main(){ scanf("%s",s+1);scanf("%d",&k);len=strlen(s+1); for (int i=1;i<=len;i++) if (len-i+1>k*2){ for (int p=i;p<=len;p++) ss[p-i+1]=s[p]; j=0; for (int p=2;p<=len-i+1;p++){ while (j&&ss[j+1]!=ss[p]) j=fail[j]; if (ss[j+1]==ss[p]) j++; fail[p]=j; } j=0; for (int p=1;p<=len-i+1;p++){ while (j&&ss[j+1]!=ss[p]) j=fail[j]; if (ss[j+1]==ss[p]) j++; int jj=j; while (jj&&(jj<<1>=p)) jj=fail[jj]; ans+=jj>=k; } // cout<<i<<' '<<ans<<endl; } cout<<ans<<endl; }