hdu2328——Corporate Identity（后缀数组练习）

今天继续后缀数组的练习，在做一道题时花了较多的时间，主要是一些细节上的问题，也是后缀数组题目常出现的一些细节问题，做完这题让我对后缀数组的使用更加熟悉了。

链接：http://acm.hdu.edu.cn/showproblem.php?pid=2328

相关细节在代码中体现：

//sa存的是排好序的后缀序号，Rank存的是每个后缀序号对应的排名
//用后缀数组时，一定要注意每个数组存的内容，比如说Map存的是每个后缀序号对应的串序号，如果你
//获得了一个后缀序号的排名i，想知道这个后缀序号对应的串序号时，就应该先通过sa[i]获得后缀序号
//在放入到Map数组中获得其对应的串序号。举这个例子就是为了说明做这种题目一定要注意区分后缀序号
//和后缀序号的排名，别搞混了
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN=900000;
int str[MAXN],sa[MAXN],height[MAXN],a[MAXN],b[MAXN],c[MAXN],Rank[MAXN];
char temp[205],ansStr[205];
bool flag[4005];
int Map[MAXN],index;
int n;
void char2int(char *s,int *top,int n,int *Min)
{
int len=strlen(s);
if((*Min)>len)
(*Min)=len;
for(int i=0;i<len;i++)
{
Map[(*top)]=n;//Map存的是后缀序号对应的串序号
str[(*top)++]=(int)s[i]+1;
}
}
void DA(int top,int m)
{
int *x=a,*y=b;
for(int i=0;i<m;i++)
c[i]=0;
for(int i=0;i<top;i++)
c[x[i]=str[i]]++;
for(int i=1;i<m;i++)
c[i]+=c[i-1];
for(int i=top-1;i>=0;i--)
sa[--c[x[i]]]=i;
for(int k=1;k<=top;k*=2)
{
int p=0;
for(int i=top-k;i<top;i++)
y[p++]=i;
for(int i=0;i<top;i++)
if(sa[i]>=k)
y[p++]=sa[i]-k;
for(int i=0;i<m;i++)
c[i]=0;
for(int i=0;i<top;i++)
c[x[y[i]]]++;
for(int i=1;i<m;i++)
c[i]+=c[i-1];
for(int i=top-1;i>=0;i--)
sa[--c[x[y[i]]]]=y[i];
swap(x,y);
p=1;
x[sa[0]]=0;
for(int i=1;i<top;i++)
x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?p-1:p++;
if(p>=top)
break;
m=p;
}
}
void getHeight(int top)
{
for(int i=1;i<=top;i++)//Rank数组最好在此函数中更新
Rank[sa[i]]=i;
int t=0;
for(int i=0;i<top;i++)
{
if(t>0)
t--;
int j=sa[Rank[i]-1];
while(str[i+t]==str[j+t])
t++;
height[Rank[i]]=t;//height存的是后缀序号排名对应的lcp
}
}
bool check()
{
for(int i=0;i<n;i++)
if(flag[i]==false)
return false;
return true;
}
bool isOk(int k,int top)
{
int temp;
memset(flag,false,sizeof(flag));
for(int i=1;i<=top;i++)
{
if(height[i]>=k)
{
flag[Map[sa[i]]]=true;//得到的i是排名值，而Map需要的是后缀序号，所以要用sa[i]转换
temp=sa[i];//temp使用来求起始下标的，也就是后缀序号，所以也要用sa[i]
}
else
{
if(check())
{
index=temp;
return true;
}
else
{
memset(flag,false,sizeof(flag));
flag[Map[sa[i]]]=true;
}
}
}
return false;
}
void getAns(int k)
{
int top=0;
for(int i=index;i<index+k;i++)//获得子串
ansStr[top++]=str[i]-1;
ansStr[top]='\0';
}
int main()
{
while(1)
{
scanf("%d",&n);
if(!n)
break;
int top=0,Min=250;
for(int i=0;i<n;i++)
{
scanf("%s",temp);
char2int(temp,&top,i,&Min);
if(i!=n-1)
{
Map[top]=4002;
str[top++]=150+i;
}
}
str[top]=0;
DA(top+1,150+n);
getHeight(top);
int ans=0;
for(int i=1;i<=Min;i++)
{
if(isOk(i,top))
ans=i;
else
break;
}
if(ans>0)
{
getAns(ans);
printf("%s\n",ansStr);
}
else
printf("IDENTITY LOST\n");
}
return 0;
}

今天做这题时，就是在一些地方到底要不要用sa数组上纠结了半天。