Tyvj 1860 后缀数组 [Suffix Array 裸题]

一句话，输出文本串的sa数组和height数组

要注意细节！！！

注意rank下标 0~lens-1 ；而 sa下标 1~lens ；height下标为2~lens

所以带入da() 时 lens要+1！！

还有在读入时注意不要使用０，从１开始！！　否则后面判断相等时可能会dead　loop！！

另外坑father的地方！！ 用printf直接T 但是用cout却不会？？！！！

yyr大神说可以输出优化，那么就只用了62ms咯

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<string>
#include<iomanip>
#include<ctime>
#include<climits>
#include<cctype>
#include<algorithm>
#ifdef WIN32
#define AUTO "%I64d"
#else
#define AUTO "%lld"
#endif
using namespace std;
#define smax(x,tmp) x=max((x),(tmp))
#define smin(x,tmp) x=min((x),(tmp))
#define maxx(x1,x2,x3) max(max(x1,x2),x3)
#define minn(x1,x2,x3) min(min(x1,x2),x3)
const int INF=0x3f3f3f3f;
const int maxn = 200005;
int t1[maxn],t2[maxn],sa[maxn],seq[maxn],cnt[maxn];
inline bool equal(int temp[],int i,int len)
{
return temp[sa[i-1]]==temp[sa[i]] && temp[sa[i-1]+len]==temp[sa[i]+len];
}
void da(int lens,int lim)
{
int *x=t1,*y=t2;
for(int i=0;i<lim;i++) cnt[i]=0;
for(int i=0;i<lens;i++) cnt[ x[i]=seq[i] ]++;
for(int i=1;i<lim;i++) cnt[i] += cnt[i-1];
for(int i=lens-1;i>=0;i--) sa[--cnt[x[i]]] = i;
for(int k=1;k<=lens;k<<=1)
{
int p = -1;
for(int i=lens-k;i<lens;i++) y[++p] = i;
for(int i=0;i<lens;i++) if(sa[i]>=k) y[++p] = sa[i]-k;
for(int i=0;i<lim;i++) cnt[i]=0;
for(int i=0;i<lens;i++) cnt[x[y[i]]]++;
for(int i=1;i<lim;i++) cnt[i] += cnt[i-1];
for(int i=lens-1;i>=0;i--) sa[--cnt[x[y[i]]]] = y[i];
swap(x,y); p=1;
x[sa[0]] = 0;
for(int i=1;i<lens;i++)
x[sa[i]] = equal(y,i,k)? p-1 : p++;
if(p>=lens) break;
lim = p;
}
}
int rank[maxn],height[maxn];
void get_height(int lens)
{
int k = 0;
for(int i=1;i<=lens;i++) rank[sa[i]] = i;
for(int i=0;i<lens;i++)
{
if(k) k--;
int j = sa[rank[i]-1];
while(seq[i+k]==seq[j+k]) k++;
height[rank[i]] = k;
}
}
int ans[20];
int tot;
inline void readout(int x)
{
if(!x)
{
putchar('0');
return;
}
tot=-1;
while(x) ans[++tot]=x%10,x/=10;
while(~tot) putchar(ans[tot--]+'0');
}
int main()
{
freopen("sa.in","r",stdin);
freopen("sa.out","w",stdout);
int lens = 0 , lim = 0;
char ch = getchar();
while(ch>='a' && ch<='z')
{
seq[lens++] = ch - 'a' + 1;
smax(lim , ch-'a'+1);
ch = getchar();
}
da(lens+1,lim+1);
get_height(lens);
for(int i=1;i<lens;i++) readout(sa[i]+1),putchar(' ');
readout(sa[lens]+1),putchar('\n');
for(int i=1;i<lens;i++) readout(height[i]),putchar(' ');
readout(height[lens]),putchar('\n');
return 0;
}

留个链接：

http://blog.csdn.net/imutzcy/article/details/13631879

http://www.cnblogs.com/Lyush/p/3233573.html