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

bzoj3620 似乎在夢中見過的樣子 kmp

..這題其實挺好想的,然而我想到正解以後,覺得不能過,然後摒棄了,打了個SA調了半天才發現位置不同才算不同否則全部相同。。
而且我正解時間複雜度分析錯了,,以為是N^2結果實現以後才發現隨便搞一搞就能線性。。
正解就是kmp暴力,,列舉左端點然後直接kmp找出所有的A,然後看看B和A符不符合條件就行了。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std; int n,m; const int N=1e5+5; char s[N]; int next[N],g[N]; int ans; int main() { scanf("%s%d",s+1,&m); n=strlen(s+1); int len=n-m*2; int k; fo(i,1,len) { next[i]=k=i-1; g[i]=i; fo(j,i+1,n) { while (k!=i-1&&s[j]!=s[k+1
])k=next[k]; if (s[j]==s[k+1])k++; next[j]=k; g[j]=(k<i+m-1)?j:g[k]; if (g[j]<((i+j)>>1))ans++; } } printf("%d\n",ans); return 0; }