【BZOJ 3620】 3620: 彷佛在夢中見過的樣子 （KMP）

3620: 彷佛在夢中見過的樣子

Time Limit: 15 Sec  Memory Limit: 128 MB
Submit: 755  Solved: 445

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

 

 

【分析】

　　作這題的時候並不知道資瓷N^2的KMP。。。

　　其實N^2的KMP還挺容易打錯的，由於根節點不能是0，是st-1。中間有幾個判斷都要注意。

　　直接枚舉起點。而後作一遍KMP。

　　詢問的時候看看nt是否符合。首先要長於k，其次不能相交且要空出一個位置。

　　假設先後綴匹配部分長度是p，枚舉到右端點爲j。

　　則2*p<j-i+1,p>=k

　　p一開始是nt，而後一直nt。可是這樣暴就O(n^3)了會超時的。

　　用一個g數組記錄，他到他的nt中，p大於等於k的最小值便可。

 

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 #include<algorithm>
 6 using namespace std;
 7 #define Maxn 15010
 8 #define INF 0xfffffff
 9 
10 char s[Maxn];
11 int l,k,nt[Maxn],g[Maxn];
12 
13 void KMP(int st)
14 {
15     nt[st]=st-1;g[st-1]=INF;
16     g[st]=k==1?1:INF;
17     for(int p=st-1,i=st+1;i<=l;i++)
18     {
19         while(s[i]!=s[p+1]&&p>=st) p=nt[p];
20         if(s[i]==s[p+1]) p++;
21         nt[i]=p;
22         g[i]=INF;
23         if(i-st+1>=k) g[i]=i-st+1;
24         g[i]=min(g[i],g[nt[i]]);
25     }
26 }
27 
28 int ans=0;
29 void ffind(int st)
30 {
31     KMP(st);
32     for(int i=st;i<=l;i++)
33     {
34         if(2*g[nt[i]]<i-st+1) ans++;
35     }
36 }
37 
38 int main()
39 {
40     scanf("%s",s+1);l=strlen(s+1);
41     scanf("%d",&k);
42     for(int i=1;i<=l;i++) ffind(i);
43     printf("%d\n",ans);
44     return 0;
45 }

View Code

 

2017-04-25 11:48:23