1. 程式人生 > >【bzoj3620】 似乎在夢中見過的樣子 KMP

【bzoj3620】 似乎在夢中見過的樣子 KMP

一開始看到資料範圍是15000就往後綴陣列上想,想了好久沒想出來怎麼處理同一子串的問題,後來看了看題解才發現O(n^2)就可以過,用KMP類似NOI2014動物園的演算法一樣。列舉左端點,對於每個右端點處理出以右端點為結尾最大長度使得從左端點開始的字首等於以右端點結束的字尾,即next陣列。然後一直往前跳,直到長度小於子串長度的一半為止。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 20010

using namespace std;

int next[maxn];
char s[maxn];
int n,k,ans;

void KMP(int p)
{
	for (int i=1;i<=n;i++) next[i]=p-1;
	for (int i=p+1;i<=n;i++)
	{
		int j=next[i-1];
		while (j!=p-1 && s[j+1]!=s[i]) j=next[j];
		if (s[j+1]==s[i]) j++;
		next[i]=j;
	}
	int j=next[p];
	for (int i=p+1;i<=n;i++)
	{
		while (j!=p-1 && s[j+1]!=s[i]) j=next[j];
		if (s[j+1]==s[i]) j++;
		while ((j-p+1)*2>=(i-p+1)) j=next[j];
		if (j-p+1>=k) ans++;
	}
}

int main()
{
	scanf("%s",s+1);
	scanf("%d",&k);
	n=strlen(s+1);
	for (int i=1;i<=n;i++)
	  KMP(i);
	printf("%d\n",ans);
	return 0;
}