hdu5442 Favorite Donut 后缀数组 长春网赛

      wa从一点到晚上11点没停过，也不知道为什么错，第二天换了个思路做，终于过了。这题还是有点问题的，数据有点水，我看到有人贴的代码baabbaab这组数据是4 0，明显错的，但是却可以过。

下面的是我第一次写的代码，也不知道到底哪里错了，能想到的样例都过了。

正的直接后缀数组求出即可。反着的我还用height注意细节。

希望大牛们指出错误，我真的找了几个小时都找不出。

#include <stdio.h>
#include <string.h>
#define maxn 51000
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];}
int min(int x,int y)
{return x<y?x:y;}
int max(int x,int y)
{return x>y?x:y;}
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 r[maxn],sa[maxn];
char s[maxn],st[maxn];
int main()
{
    int i,j,t,n;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        if(n==0)
        {
            printf("0 0\n");
            continue;
        }
        scanf("%s",s);
        int len=n;
        for(i=0;i<n;i++)
            s[len++]=s[i];
        for(i=0;i<len;i++)
            r[i]=s[i];
        r[len]=0;
        da(r,sa,len+1,125);
        calheight(r,sa,len);
        int setone=0;
        for(i=1;i<=len;i++)
        {
            if(len-sa[i]-1<n)
                continue;
            else if(height[i]>=n)
                setone=min(sa[i-1],sa[i]);
            else
            {
                setone=sa[i];
            }
        }
        setone%=n;
        //printf("%d\n",setone);
        int fl=0;
        for(i=len-1;i>=0;i--)
            st[fl++]=s[i];
        for(i=0;i<len;i++)
            r[i]=st[i];
        r[len]=0;
        da(r,sa,len+1,125);
        calheight(r,sa,len);
        int settwo=0;
        for(i=1;i<=len;i++)
        {
            if(len-sa[i]-1<n)
                continue;
            else if(height[i]>=n)
                settwo=max(sa[i-1],sa[i]);
            else
            {
                settwo=sa[i];
            }
        }
        //printf("%d\n",settwo);
        int flag=0;
        for(i=0;i<n;i++)
        {
            if(s[setone+i]==st[settwo+i])
                continue;
            else if(s[setone+i]>st[settwo+i])
            {
                flag=1;
                break;
            }
            else 
            {
                flag=2;
                break;
            }
        }
        settwo=n-settwo%n-1;
        if(flag==1)
        {
            printf("%d 0\n",setone+1);
        }
        else if(flag==2)
        {
            printf("%d 1\n",settwo+1);
        }
        else
        {
            if(settwo>=setone)
                printf("%d 0\n",setone+1);
            else printf("%d 1\n",settwo+1);
        }
    }
}

下面是ac代码，反向通过先在末尾加未出现过的最大字母，再多添加0（为了使0位附0），那么就能避免上面代码出现的情况，即满足条件的在sa[]里面的中间部分，但是不能确定后面是否还有，添加字母后，就能保证能够满足，这样只要在[0，原字符串长度）里找到最大的rank的位置就好。

#include <stdio.h>
#include <string.h>
#define maxn 101000
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];}
int min(int x,int y)
{return x<y?x:y;}
int max(int x,int y)
{return x>y?x:y;}
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 r[maxn],sa[maxn];
char s[maxn],st[maxn];
int main()
{
    int i,j,t,n;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        scanf("%s",s);
        memset(r,0,sizeof(r));
        int len=n;
        for(i=0;i<n;i++)
            s[len++]=s[i];
        for(i=0;i<len;i++)
            r[i]=s[i];
        r[len]=0;
        da(r,sa,len+1,125);
        calheight(r,sa,len);
        int setone=0;
        int maxx=-1;
        for(i=0;i<n;i++)
        {
            if(maxx<Rank[i])
            {
                maxx=Rank[i];
                setone=i;
            }
        }
        //printf("%d\n",setone);
        memset(r,0,sizeof(r));
        int fl=0;
        for(i=len-1;i>=0;i--)
            st[fl++]=s[i];
        for(i=0;i<len;i++)
            r[i]=st[i];
        r[len++]=123;//添加一个最大的
        r[len]=0;//0位无字符
        da(r,sa,len+1,125);
        calheight(r,sa,len);
        int settwo=0;
        maxx=-1;
        for(i=0;i<n;i++)
        {
            if(maxx<Rank[i])
            {
                maxx=Rank[i];
                settwo=i;
            }
        }
        //printf("%d\n",settwo);
        int flag=0;
        for(i=0;i<n;i++)
        {
            if(s[setone+i]==st[settwo+i])
                continue;
            else if(s[setone+i]>st[settwo+i])
            {
                flag=1;
                break;
            }
            else 
            {
                flag=2;
                break;
            }
        }
        settwo=n-settwo%n-1;
        if(flag==1)
        {
            printf("%d 0\n",setone+1);
        }
        else if(flag==2)
        {
            printf("%d 1\n",settwo+1);
        }
        else
        {
            if(settwo>=setone)
                printf("%d 0\n",setone+1);
            else printf("%d 1\n",settwo+1);
        }
    }
    return 0;
}