1. 程式人生 > >bzoj 3620: 似乎在夢中見過的樣子

bzoj 3620: 似乎在夢中見過的樣子

動物 data cstring ring string printf pre efi 附近

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

首先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);	
}

  

bzoj 3620: 似乎在夢中見過的樣子