uoj#35. 接尾辞ソート接尾辞配列


uoj#35. 接尾辞ソート裸の接尾辞配列・・・2回叩いて・・・QQQがlrjコードとあるAC神を参考にして書いたコードを忘れないように・・・
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
const int N=200005;
int n;
int sa[N],c[N],t1[N],t2[N];
int rank[N],height[N];
char s[N];
void buildsa(int m){
    int i,*x=t1,*y=t2;
    for(i=1;i<=m;i++) c[i]=0;
    for(i=1;i<=n;i++) c[x[i]=s[i]]++;
    for(i=1;i<=m;i++) c[i]+=c[i-1];
    for(i=n;i>=1;i--) sa[c[x[i]]--]=i;
    for(int k=1;k<=n;k<<=1){
        int p=0;
        for(i=n-k+1;i<=n;i++) y[++p]=i;
        for(i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;
        for(i=1;i<=m;i++) c[i]=0;
        for(i=1;i<=n;i++) c[x[y[i]]]++;
        for(i=1;i<=m;i++) c[i]+=c[i-1];
        for(i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i];
        swap(x,y);
        x[sa[1]]=1;p=1;
        for(i=2;i<=n;i++)
            x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p:++p;
        if(p>=n) break;
        m=p;
    }
}
void getheight() {
    int i,j,k=0;
    for (i=1;i<=n;i++) rank[sa[i]]=i;
    for (i=1;i<=n;i++) {
        if (k) k--;
        j=sa[rank[i]-1];
        while (s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }
}
int main(){
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    scanf("%s",s+1);n=strlen(s+1);
    buildsa(128);
    for(int i=1;i<=n;i++) printf("%d ",sa[i]);puts("");
    getheight();
    for(int i=2;i<=n;i++) printf("%d ",height[i]);puts("");
    return 0;
}

その神妙さに注釈をつけた
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int maxn=200011;
char s[maxn];
int c[maxn],t1[maxn],t2[maxn],sa[maxn],rank[maxn],height[maxn],f[maxn][20];
int i,j,k,m,n;

//        ,         s[1..n]
//sa[i]      i      s    
//rank[i]     i       s         
//height[i]      i-1         i           

void buildsa(int m,int *sa,char *s) {
    int i,*x=t1,*y=t2;
    for (i=1;i<=m;i++) c[i]=0;
    for (i=1;i<=n;i++) c[x[i]=s[i]]++;//  ,               x[i]
    for (i=1;i<=m;i++) c[i]+=c[i-1];//    
    for (i=n;i>=1;i--) sa[c[x[i]]--]=i;//  2^0     
    for (k=1;k<=n;k<<=1) {
        //               2^k     
        int p=0;
        //       
        for (i=n-k+1;i<=n;i++) y[++p]=i;//           
        for (i=1;i<=n;i++) if (sa[i]>k) y[++p]=sa[i]-k;//sa[i]  sa[i]   ,  k           ,  ;
        //     k   。

        //       
        for (i=1;i<=m;i++) c[i]=0;//   
        for (i=1;i<=n;i++) c[x[y[i]]]++;//       
        for (i=1;i<=m;i++) c[i]+=c[i-1];//    |          
        for (i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i];//             sa             
        swap(x,y);//y      rank
        p=1;x[sa[1]]=1;
        for (i=2;i<=n;i++) x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k]?p:++p;
        if (p>=n) break;
        m=p;
    }
}

void getheight(char *s,int *sa,int *rank,int *height) {
    int i,j,k=0;
    for (i=1;i<=n;i++) rank[sa[i]]=i;//     sa    rank  
    for (i=1;i<=n;i++) {
        if (k) k--;
        j=sa[rank[i]-1];//  height[rank[i]]>=height[rank[i-1]]-1  height  
        while (s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }
}

void initrmq(int f[maxn][20],int *height) {
    int i,j;
    for (i=1;i<=n;i++) f[i][0]=height[i];
    for (j=1;1<<j<=n;j++)
        for (i=1;i<=n-(1<<j)+1;i++)
            f[i][j]=min(f[i][j-1],f[i+(1<<j>>1)][j-1]);
}

int getrmq(int f[maxn][20],int i,int j) {
    if (!i || !j) return 0;
    if (i>j) swap(i,j);
    int s=i+1,t=j,k;
    for (k=0;(1<<k)<=t-s+1;k++);
    k--;
    return min(f[s][k],f[t-(1<<k)+1][k]);
}

int main() {
    freopen("in.txt","r",stdin);
    freopen("right.txt","w",stdout);
    scanf("%s",s+1);n=strlen(s+1);
    buildsa(128,sa,s);
    getheight(s,sa,rank,height);
    for (i=1;i<=n;i++) printf("%d ",sa[i]);
    puts("");
    for (i=2;i<=n;i++) printf("%d ",height[i]);
    puts("");
    return 0;
}