【EXKMP】AHOI 2007 D1T3 灯阵


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

算法想出来了的话也没什么好解释的,弄了好久才搞懂,主要是对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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值