1. 程式人生 > >pku1743字尾樹組-4

pku1743字尾樹組-4

 http://poj.org/problem?id=1743

題意:給定n個數字,求著n個數字的一個“子串”,保證這個子串在原陣列中出現了至少2次,且這兩次不不出現重疊(而且這兩個子串不是完全相同,而是可以有一定的“跳躍”,比如1 2 3 4與7 8 9 10,前一個串的每個都元素加上6就變成後一個串了,這個也選相同,題意看完就被嚇慘了)。。。求最長的子串長度。。。

 

分析:此題為樓教主男人8題中的一題,終於也可以做1/8個男人了。。呵呵。。

其實熟悉字尾樹組的童鞋應該清楚,其實這題不算難。。。但事實上如果不是luohuiqian在論文中說這題是字尾樹組我真的不知道怎麼處理,因為兩個串不是完全相同,而是可以有一定的“跳躍”。。。其實想想應該是可以想到的,無論怎麼跳,如果兩個串是這樣“相同”,則其兩兩之間的差值。。。不解釋了。。呵呵。。然後就轉化成不重疊的最長公共子串問題了,字尾樹組+二分答案即可。。。。。老是二分啊。。。。

 

程式碼:

#include<stdio.h>
#include<string.h>

#define maxn 20011
int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
int cmp(int *r,int a,int b,int l)
{
	return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int *sa,int n,int m)
{
     int i,j,p,*x=wa,*y=wb,*t;
     for(i=0;i<m;i++) ws[i]=0;
     for(i=0;i<n;i++) ws[x[i]=r[i]]++;
     for(i=1;i<m;i++) ws[i]+=ws[i-1];
     for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
     for(j=1,p=1;p<n;j*=2,m=p)
     {
       for(p=0,i=n-j;i<n;i++) y[p++]=i;
       for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
       for(i=0;i<n;i++) wv[i]=x[y[i]];
       for(i=0;i<m;i++) ws[i]=0;
       for(i=0;i<n;i++) ws[wv[i]]++;
       for(i=1;i<m;i++) ws[i]+=ws[i-1];
       for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
       for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
       x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
     return;
}
int rank[maxn],height[maxn];
void calheight(int *r,int *sa,int n)
{
     int i,j,k=0;
     for(i=1;i<=n;i++) rank[sa[i]]=i;
     for(i=0;i<n;height[rank[i++]]=k)
     for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
     return;
}

int n, a[maxn], sa[maxn];

int check(int mid)
{
	int i, j, mn, mx;
	for(i=1; i<=n; i=j)
	{
		while(i<=n && height[i]<mid)
			i++;
		j=i;
		mn = n;
		mx = 0;
		if(i==n+1)
			break;
		if(sa[i-1]>mx)
			mx = sa[i-1];
		if(sa[i-1]<mn)
			mn = sa[i-1];
		while(j<=n && height[j]>=mid)
		{
			if(sa[j]>mx)
				mx = sa[j];
			if(sa[j]<mn)
				mn = sa[j];
			j++;
		}
		if(mx-mn>=mid)
			return 1;
	}
	return 0;
}

int main()
{
	int i, l, r, mid, ans;
	while(scanf("%d", &n)!=EOF)
	{
		if(n==0)
			break;
		if(n<=5)
		{
			for(i=0; i<n; i++)
				scanf("%d", &a[i]);
			printf("0\n");
			continue;
		}
		scanf("%d", &a[0]);
		for(i=1; i<n; i++)
		{
			scanf("%d", &a[i]);
			a[i-1] = a[i]-a[i-1]+89;
		}
		n--;
		a[n] = 0;
		da(a, sa, n+1, 180);
		calheight(a, sa, n);
		l=0, r=n-1;
		while(l<=r)
		{
			mid = (l+r)>>1;
			if(check(mid))
				l = mid+1;
			else
				r = mid-1;
		}
		ans = r;
		if(ans>=4)
			printf("%d\n", ans+1);
		else
			printf("0\n");
	}

	return 0;
}