[POJ2774] 롱롱 메시지 접미사 로봇

...다른 문제는 일부 코드가 맞는지 보려고 썼다.그래서 SAM을 가지고 이 문제를 다시 한 번 냈다.
코드만 붙이고,
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 101000
#define T 30
using namespace std;
char s[N],t[N];
int pa[N<<1],son[N<<1][T];
int deep[N<<1],cnt,root,last;
inline int Newnode(int _deep){deep[++cnt]=_deep;return cnt;}
inline void SAM(int alp)
{
	int u=last;
	last=Newnode(deep[last]+1);
	while(u&&!son[u][alp])son[u][alp]=last,u=pa[u];
	if(!u)pa[last]=root;
	else {
		int v=son[u][alp];
		if(deep[v]==deep[u]+1)pa[last]=v;
		else {
			int nv=Newnode(deep[u]+1);
			memcpy(son[nv],son[v],sizeof son[v]);
			pa[nv]=pa[v],pa[v]=pa[last]=nv;
			while(u&&son[u][alp]==v)son[u][alp]=nv,u=pa[u];
		}
	}
}
int lena,lenb;
int solve(int n)
{
	int i,x,v,ret=0,temp=0;
	x=root=last=Newnode(0);
	for(i=1;i<=lena;i++)SAM(s[i]-'a');
	for(i=1;i<=lenb;i++)
	{
		if(son[x][t[i]-'a'])x=son[x][t[i]-'a'],temp++;
		else {
			while(x&&son[x][t[i]-'a']==0)x=pa[x];
			if(!x)x=root,temp=0;
			else temp=deep[x]+1,x=son[x][t[i]-'a'];
		}
		ret=max(ret,temp);
	}
	return ret;
}
int main()
{
//	freopen("test.in","r",stdin);
	int i,j,n,ans=0;
	scanf("%s %s",s+1,t+1);
	lena=strlen(s+1),lenb=strlen(t+1);
	printf("%d
",solve(n)); return 0; }

좋은 웹페이지 즐겨찾기