bzoj3620 似乎在夢中見過的樣子 kmp
阿新 • • 發佈:2019-01-06
..這題其實挺好想的,然而我想到正解以後,覺得不能過,然後摒棄了,打了個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;
}