【EXKMP】AHOI 2007 D1T3 灯阵

来源：互联网发布：node.js 后端还是前端编辑：程序博客网时间：2024/05/15 23:50

题目很漂亮，但不是那么好想的

算法想出来了的话也没什么好解释的，弄了好久才搞懂，主要是对ext[]的理解，目前还没明白扩展KMP的ext[]是如何得出的

用链表维护删除保证了O(n)的复杂度

//#define _TEST _TEST#include <cstdio>#include <cstring>#include <cstdlib>#include <iostream>#include <cmath>#include <algorithm>using namespace std;/************************************************Code By willinglive    Blog:http://willinglive.cf************************************************/#define rep(i,l,r) for(int i=(l),___t=(r);i<=___t;i++)#define per(i,r,l) for(int i=(r),___t=(l);i>=___t;i--)#define MS(arr,x) memset(arr,x,sizeof(arr))#define LL long long#define INE(i,u,e) for(int i=head[u];~i;i=e[i].next)inline const int read(){int r=0,k=1;char c=getchar();for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';return k*r;}/////////////////////////////////////////////////int n,a[500010],ext[500010];int pre[500010],next[500010];struct data{int pos,next;}d[500010];int head[500010],k;/////////////////////////////////////////////////void get_fail(){ext[1]=n;int i=1,t=2;while(i+1<=n && a[i]==a[i+1]) i++;ext[2]=i-1;for(i=2;i<n;i++){int p=max(t+ext[t]-1,i);int l=ext[i-t+2];if(i+l<p) ext[i+1]=l;else{int j=1;while(p+j<=n && a[p+j]==a[p-i+j]) j++;j--;ext[i+1]=p-i+j;if(t+ext[t]<i+1+ext[i+1]) t=i+1;}}/*int j=0;rep(i,2,n){while(j&&a[j+1]!=a[i]) j=fail[j];fail[i]=j+=a[j+1]==a[i];}*/}void add(int x,int pos){d[k].pos=pos; d[k].next=head[x]; head[x]=k++;}void init(){MS(head,-1);rep(i,1,n) pre[i]=i-1,next[i]=i+1;rep(i,1,n) add(ext[i],i);}int del(int pos){next[pre[pos]]=next[pos];pre[next[pos]]=pre[pos];return next[pos]-pre[pos];}/////////////////////////////////////////////////void input(){    rep(i,1,n=read()) a[i]=read();}void solve(){get_fail();init();//rep(i,1,n) printf("%d ",ext[i]); puts("");int mx=0;rep(len,1,n){for(int j=head[len-1];~j;j=d[j].next)    mx=max(mx,del(d[j].pos));if(mx<=len) printf("%d\n",len);}}/////////////////////////////////////////////////int main(){    #ifndef _TEST    freopen("light.in","r",stdin); freopen("light.out","w",stdout);    #endif    input(),solve();    return 0;}

0 0