POJ2774 Long Long Message 后缀数组（裸模板=。=）

省赛加油。

第一次做后缀数组的题，每日一题系列。

裸的后缀数组，讲两个字符串合并，中间加一个其他字符隔断，找出height最大的，且在两个字符串中。

不知道为什么用了两个模板全挂了，连罗穗骞的也不能幸免，明天早仔细研究，智商还是跟不上啊。

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

#include <iostream>
using namespace std;
const int MAX=210000;
int *rank,r[MAX],sa[MAX],height[MAX];
int wa[MAX],wb[MAX],wm[MAX];
char s[MAX];

bool cmp(int *r,int a,int b,int l){
    return r[a] == r[b] && r[a+l] == r[b+l];
}

void makesa(int *r,int *sa,int n,int m){
    int *x=wa,*y=wb,*t;
    for(int i=0;i<m;++i)wm[i]=0;
    for(int i=0;i<n;++i)wm[x[i]=r[i]]++;
    for(int i=1;i<m;++i)wm[i]+=wm[i-1];
    for(int i=n-1;i>=0;--i)sa[--wm[x[i]]]=i;
    for(int i=0,j=1,p=0;p<n;j=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<m;++i)wm[i]=0;
        for(i=0;i<n;++i)wm[x[y[i]]]++;
        for(i=1;i<m;++i)wm[i]+=wm[i-1];
        for(i=n-1;i>=0;--i)sa[--wm[x[y[i]]]]=y[i];
        for(t=x,x=y,y=t,i=p=1,x[sa[0]]=0;i<n;++i){
            x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
        }
    }
    rank=x;
}

void calheight(int *r,int *sa,int n){
    for(int i=0,j=0,k=0;i<n;height[rank[i++]]=k){
        for(k?--k:0,j=sa[rank[i]-1];r[i+k] == r[j+k];++k);
    }
}

int main()
{
    int ans,y,n;
        gets(s);
            ans=0,n=0;
            for(n=0;s[n] != '\0';++n)r[n]=s[n];
        r[y=n]='#';
        scanf("%s",s+n+1);//将输入的两个字符串连接在一起,中间用'#'隔开
        for(++n;s[n] != '\0';++n)r[n]=s[n];
        r[n+1]=0;
    makesa(r,sa,n+1,256);
    calheight(r,sa,n);
    for(int i=1;i<n;i++)
    {
        if((sa[i]<y&&sa[i-1]>y)||(sa[i]>y&&sa[i-1]<y))
            ans=max(ans,height[i]);
      //  printf("%d\n",height[i]);
    }
    printf("%d\n",ans);

    return 0;
}

好吧，刚发完就发现问题了，看代码==

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
#define rep(i,n) for(int i = 0;i < n; i++)
const int size  = 210000;
int rk[size],sa[size],height[size],w[size],wa[size],res[size];
void getSa (int len,int up) {
	int *k = rk,*id = height,*r = res, *cnt = wa;
	rep(i,up) cnt[i] = 0;
	rep(i,len) cnt[k[i] = w[i]]++;
	rep(i,up) cnt[i+1] += cnt[i];
	for(int i = len - 1; i >= 0; i--) {
		sa[--cnt[k[i]]] = i;
	}
	int d = 1,p = 0;
	while(p < len){
		for(int i = len - d; i < len; i++) id[p++] = i;
		rep(i,len)	if(sa[i] >= d) id[p++] = sa[i] - d;
		rep(i,len) r[i] = k[id[i]];
		rep(i,up) cnt[i] = 0;
		rep(i,len) cnt[r[i]]++;
		rep(i,up) cnt[i+1] += cnt[i];
		for(int i = len - 1; i >= 0; i--) {
			sa[--cnt[r[i]]] = id[i];
		}
		swap(k,r);
		p = 0;
		k[sa[0]] = p++;
		rep(i,len-1) {
			if(sa[i]+d < len && sa[i+1]+d <len &&r[sa[i]] == r[sa[i+1]]&& r[sa[i]+d] == r[sa[i+1]+d])
				k[sa[i+1]] = p - 1;
			else k[sa[i+1]] = p++;
		}
		if(p >= len) return ;
		d *= 2,up = p, p = 0;
	}
}
void getHeight(int len) {
	rep(i,len) rk[sa[i]] = i;
	height[0] =  0;
	for(int i = 0,p = 0; i < len - 1; i++) {
		int j = sa[rk[i]-1];
		while(i+p < len&& j+p < len&& w[i+p] == w[j+p]) {
			p++;
		}
		height[rk[i]] = p;
		p = max(0,p - 1);
	}
}
char s[size];
int main()
{
    int ans,y,n;
        gets(s);
            ans=0,n=0;
            for(n=0;s[n] != '\0';++n)w[n]=s[n];
        w[y=n]='#';
        scanf("%s",s+n+1);//将输入的两个字符串连接在一起,中间用'#'隔开
        for(++n;s[n] != '\0';++n)w[n]=s[n];
        w[n+1]=0;
    getSa(n+1,256);
    getHeight(n+1);    //传的是n+1！！！！！！
    for(int i=1;i<n;i++)
    {
        if((sa[i]<y&&sa[i-1]>y)||(sa[i]>y&&sa[i-1]<y))
            ans=max(ans,height[i]);
      //  printf("%d\n",height[i]);
    }
    printf("%d\n",ans);

    return 0;
}

安了