有些類似noi2014的動物園,也是對于KMP算法的一個應用,思想就是枚舉前綴然后預先留出k的位置,對于自身KMP,當next數組>k時就計入答案
#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<algorithm>using namespace std;#define N 15005char 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;}
|
新聞熱點
疑難解答