bzoj 1396 识别子串

1396: 识别子串

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 396   Solved: 252
[ Submit][ Status][ Discuss]

Description

Input

一行,一个由小写字母组成的字符串S,长度不超过10^5

Output

L行,每行一个整数,第i行的数据表示关于S的第i个元素的最短识别子串有多长.

Sample Input

agoodcookcooksgoodfood

Sample Output

1
2
3
3
2
2
3
3
2
2
3
3
2
1
2
3
3
2
1
2
3

4

【分析】

吕老板说SAM比sa更无脑的时候我的内心是绝望的(到底哪里无脑了...)

——题解摘自 Candy?

建SAM，|Right|=1的可以作为识别子串哦

|Right(s)|=1 出现位置就是Max(s)

考虑它可以作为哪些位置的识别子串

令r=Max(s),l=Max(s)-Max(fa)

[1,l-1]可以，贡献为r-i+1

[l,r]可以，贡献为r-l+1

用两颗线段树就行了

肝了两节课啊...疲劳

【代码】

//bzoj 1396 识别子串
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#define inf 1e9+7
#define ll long long
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=200005;
char s[mxn];
bool bo[mxn]; //有无pre反向边 
int n,m,p,q,np,nq,len,tot,root;
int step[mxn],pre[mxn],son[mxn][30];
struct segtree  //A:维护min(key[i]),B:维护min(key[i]+i)
{
	struct node
	{
		int l,r,mn;  //mn:只用来pushdown 
		node():mn(inf){}
	}t[mxn<<1];
	inline void build(int l,int r,int num)
	{
		t[num].l=l,t[num].r=r;
		if(l==r) return;
		int mid=l+r>>1;
		build(l,mid,num<<1);
		build(mid+1,r,num<<1|1);
	}
	inline void move(int num,int c)
	{
		t[num].mn=min(t[num].mn,c);
	}
	inline void pushdown(int num)
	{
		if(t[num].mn!=inf)
		  move(num<<1,t[num].mn),move(num<<1|1,t[num].mn);
		t[num].mn=inf;
	}
	inline void change(int num,int L,int R,int c)
	{
		if(L<=t[num].l && t[num].r<=R)
		{
			move(num,c);return;
		}
		pushdown(num);
		int mid=t[num].l+t[num].r>>1;
		if(L<=mid) change(num<<1,L,R,c);
		if(R>mid) change(num<<1|1,L,R,c);
	}
	inline int query(int num,int pos)
	{
		if(t[num].l==t[num].r) return t[num].mn;
		int mid=t[num].l+t[num].r>>1;
		pushdown(num);
		if(pos<=mid) return query(num<<1,pos);
		else return query(num<<1|1,pos);
	}
}A,B;
inline void sam()
{
	int i,j;
	scanf("%s",s+1);
	len=strlen(s+1);
	root=tot=np=1;
	fo(i,1,len)
	{
		int c=s[i]-'a'+1;
		p=np;
		step[np=(++tot)]=step[p]+1;
		while(p && !son[p][c])
		  son[p][c]=np,p=pre[p];
		if(!p)
		{
			pre[np]=root;
			continue;
		}
		q=son[p][c];
		if(step[q]==step[p]+1)
		  pre[np]=q;
		else
		{
			step[nq=(++tot)]=step[p]+1;
			memcpy(son[nq],son[q],sizeof son[q]);
			pre[nq]=pre[q];
			pre[q]=pre[np]=nq;
			while(p && son[p][c]==q)
			  son[p][c]=nq,p=pre[p];
		}
	}
}
int main()
{
	int i,j;
	sam();
	fo(i,1,tot) bo[pre[i]]=1;
	A.build(1,len,1),B.build(1,len,1);
	fo(i,1,tot)
	  if(!bo[i])
	  {
	  	  int l=step[i]-step[pre[i]],r=step[i];
	  	  B.change(1,1,l-1,r+1);
	  	  A.change(1,l,r,r-l+1);
	  }
	fo(i,1,len)
	  printf("%d\n",min(A.query(1,i),B.query(1,i)-i));
	return 0;
}