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

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

有些類似noi2014的動物園,也是對於KMP演算法的一個應用,思想就是列舉字首然後預先留出k的位置,對於自身KMP,當next陣列>k時就計入答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>

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++)//列舉左端點,對每一個左端點做KMP { for (int j=0,i=2;i+p<l;i++)//處理next(f)陣列 { 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++)//類似noi2014的動物園
{ 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++;//當前綴與字尾都>=k即j>=k時並且<=i>>1時計入答案 } } cout<<ans; return 0; }