UVa11107 Lifeform

题目链接

UVa11107 Lifeform

题解

学了高大上的后缀数组但是还是不怎么会用,关键是版子最开始打错了,害得我必须再重新理解一遍才能找出bug,开始以为是memset(sa,0,sizeof(sa))这句少写了但是并不是这样.话说字符串的题特别容易RE.

这道题二分答案然后后缀数组,先将字符串合并,每个串后加一个奇怪的不相同的字符,二分长度,然后在height数组中去找,每次当当前height值小于ans的时候再开一段,这时检查前一段是否符合要求.具体方法先用一个idx[i]表示字符串中第i个属于第几个字符串然后用flag[i]来表示有没有在第i个字符串中出现,最后统计一下flag==1的个数,若>n/2则return true;复杂度更低的O(n*L)正准备学.

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=100*2+10;
const int MAXLEN=10000000*2;
int idx[MAXLEN],n;
struct SuffixArray{
int s[MAXLEN];
int sa[MAXLEN],height[MAXLEN],rank[MAXLEN],n;
int t[MAXLEN*2],t2[MAXLEN*2];
long long cnt[MAXLEN];
SuffixArray(){n=0;}
void clear(){n=0;}
void build_sa(int m){
int i,*x=t,*y=t2;
for(i=0;i<m;i++) cnt[i]=0;
for(i=0;i<n;i++) cnt[x[i]=s[i]]++;
for(i=1;i<m;i++) cnt[i]+=cnt[i-1];
for(i=n-1;i>=0;i--) sa[--cnt[x[i]]]=i;//血的教训,妈蛋这个等号害了我一整天
for(int k=1,p=0;k<n;k <<=1)
{
p=0;
for(i=n-k;i<n;i++) y[p++]=i;
for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
for(i=0;i<m;i++) cnt[i]=0;
for(i=0;i<n;i++) cnt[x[y[i]]]++;
for(i=1;i<m;i++) cnt[i]+=cnt[i-1];
for(i=n-1;i>=0;i--) sa[--cnt[x[y[i]]]]=y[i];
swap(x,y);
p=1;x[sa[0]]=0;
for(i=1;i<n;i++)
if(y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k])
x[sa[i]]=p-1;
else x[sa[i]]=p++;
if(p>=n)
break;
m=p;
}
}
void build_height()
{
int k=0;
for(int i=0;i<n;i++) rank[sa[i]]=i;
for(int i=0;i<n;i++)
{
if(k) k--;
if(!rank[i]) continue;
int j=sa[rank[i]-1];
while(s[i+k]==s[j+k]) k++;
height[rank[i]]=k;
}
}
}SA;

inline void add(int ch,int id){
idx[SA.n]=id;
SA.s[SA.n++]=ch;
}

int flag[maxn];
inline int check(int ans){
int check_clock=1;
memset(flag,0,sizeof(flag));
for(int i=1;i<SA.n;i++)
{
if(SA.height[i]>=ans)
{
flag[idx[SA.sa[i]]]=check_clock;
flag[idx[SA.sa[i-1]]]=check_clock;
}
else {
int cnt=0;
for(int j=0;j<n;j++)
if(flag[j]==check_clock) cnt++;
if(cnt>n/2) return true;
flag[idx[SA.sa[i]]]=++check_clock;
}
}
return false;
}

inline void print_ans(int l,int r){
for(int i=l;i<=r;i++)
printf("%c",SA.s[i]+'a'-1);
printf("\n");
}

inline void print(int ans){
int check_clock=1;
memset(flag,0,sizeof(flag));
for(int i=1;i<SA.n;i++)
{
if(SA.height[i]>=ans)
{
flag[idx[SA.sa[i]]]=check_clock;
flag[idx[SA.sa[i-1]]]=check_clock;
}
else {
int cnt=0;
for(int j=0;j<n;j++)
if(flag[j]==check_clock) cnt++;
if(cnt>n/2) print_ans(SA.sa[i-1],SA.sa[i-1]+ans-1);
flag[idx[SA.sa[i]]]=++check_clock;
}
}
}
char s[1000+10];
int kase=0;
int main()
{
int maxlen;
while(scanf("%d",&n)==1&&n)
{
if(kase++) printf("\n");
SA.clear();
maxlen=0;
for(int i=0;i<n;i++){
scanf("%s",s);
int l=strlen(s);
maxlen=max(maxlen,l);
for(int j=0;j<l;j++) add(s[j]-'a'+1,i);
add(100+i,n);
}
if(n==1)
{
printf("%s\n",s);continue;
}
SA.build_sa(101+n);
SA.build_height();
int l=1,r=maxlen,ans=0;
while(l<=r)
{
int mid=((l+r)>>1);
if(check(mid)) ans=mid,l=mid+1;
else r=mid-1;
}
if(ans)print(ans);
else printf("?\n");
}
return 0;
}