【后缀数组】【spoj 694】Distinct Substrings

最新推荐文章于 2019-08-31 11:55:10 发布

willinglive

最新推荐文章于 2019-08-31 11:55:10 发布

阅读量439

点赞数

分类专栏： string

本文链接：https://blog.csdn.net/willinglive/article/details/41699553

版权

string 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

啊哈哈哈被后缀数组虐了一下午，写了三道题有两道没调处来，只有一道AC了

全都是套模板各种莫名其妙连sa[]都算不对

基本应用——求一个字符串的不重复子串的个数。

主要是对sa[]和height[]的理解

//#define _TEST _TEST
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
/************************************************
Code By willinglive    Blog:http://willinglive.cf
************************************************/
#define rep(i,l,r) for(int i=l,___t=(r);i<=___t;i++)
#define per(i,r,l) for(int i=r,___t=(l);i>=___t;i--)
#define MS(arr,x) memset(arr,x,sizeof(arr))
#define LL long long
#define INE(i,u,e) for(int i=head[u];~i;i=e[i].next)
inline const int read()
{int r=0,k=1;char c=getchar();for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;
for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';return k*r;}
/
const int m=128;
int n;
char s[1010];
int t1[1010],t2[1010],c[1010];
int sa[1010],height[1010],rk[1010];
/
void get_sa(int *x,int *y,int *sa,int n,int m)
{
	rep(i,0,m-1) c[i]=0;
	rep(i,0,n-1) c[x[y[i]]]++;
	rep(i,1,m-1) c[i]+=c[i-1];
	per(i,n-1,0) sa[--c[x[y[i]]]]=y[i];
}
void get_height(char *s,int *sa,int *height,int n)
{
	int i,j,k=0;
	rep(i,1,n) rk[sa[i]]=i;
	for(i=0;i<n;height[rk[i++]]=k)
	    for(k?k--:0,j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
}
void da(char *s,int *sa,int n)
{
	int i,j,p,m=::m,*x=t1,*y=t2,*t;
	rep(i,0,n-1) x[i]=s[i],y[i]=i;
	get_sa(x,y,sa,n,m);
	for(j=p=1;p<n;j<<=1,m=p)
	{
		p=0;
		rep(i,n-j,n-1) y[p++]=i;
		rep(i,0,n-1) if(sa[i]>=j) y[p++]=sa[i]-j;
		get_sa(x,y,sa,n,m);
		t=x;x=y;y=t;p=1;x[sa[0]]=0;
		rep(i,1,n-1) x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]?p-1:p++;
	}
	get_height(s,sa,height,n);
}
/
void input()
{
	MS(s,0); MS(sa,0); MS(height,0); MS(c,0); MS(t1,0); MS(t2,0); MS(rk,0);
    scanf("%s",s+1);
    n=strlen(s+1);
}
void solve()
{
    da(s,sa,n+1);
    int ans=0;
    rep(i,1,n) ans+=n-sa[i]-height[i]+1;
    printf("%d\n",ans);
}
/
int main()
{
    #ifndef _TEST
    freopen("std.in","r",stdin); freopen("std.out","w",stdout);
    #endif
    for(int T=read();T--;)
    input(),solve();
    return 0;
}