POJ 3294 Life Forms

标签：后缀数组

这道题搞了好久，刚开始数组开小了RE，然后是用了set就TLE，最后用数组替换set就变成了WA，最后终于发现了问题，原来if语句和else语句之间的逻辑没搞清楚

A掉之后感觉眼泪都要掉下来了

再总结一下A题的经验，做不出的题不要一直死磕，只要连续1-2个小时都没什么头绪，就可以先去干干别的事了，甚至可以去做做别的题目，给自己清空一下脑子

等到过段时间再回来做的时候，往往就能解决了

#include<stdio.h> 
#include<string.h>
#include<set>
#include<map>
#include<algorithm>
using namespace std;
#define N 201005
char s[101][2001];
int r[N],sa[N],height[N],rank[N],wa[N],wb[N],wv[N],ws[N];
int m[N];
int cmp(int *r,int a,int b,int l) {
	return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int *sa,int n,int m){
    int i,j,p,*x=wa,*y=wb;
    for(i=0;i<m;i++) ws[i]=0;
    for(i=0;i<n;i++) ws[x[i]=r[i]]++;
    for(i=1;i<m;i++) ws[i]+=ws[i-1];
    for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
    for(j=1,p=1;p<n;j<<=1,m=p){//p<n是做了常数优化，有可能提前完成排序
        for(p=0,i=n-j;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=0;i<n;i++) wv[i]=x[y[i]];
        for(i=0;i<m;i++) ws[i]=0;
        for(i=0;i<n;i++) ws[wv[i]]++;
        for(i=1;i<m;i++) ws[i]+=ws[i-1];
        for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
        swap(x,y);
        for(p=1,x[sa[0]]=0,i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
    return;
}
void calheight(int *r,int *sa,int n){
	int i,j,k=0;
	for(i=1;i<=n;i++) rank[sa[i]]=i;
	for(i=0;i<n;height[rank[i++]]=k) 
		for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
	//for(i=1;i<=n;i++) printf("%d ",height[i]);
	return;
}
int vis[1200];
int judge(int mid,int l,int n){
	int size=0;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=l;i++){
		if(height[i]>=mid){//因为下面是用的else，所以m[sa[i-1]]!=m[sa[i]]语句不能写在这个if里，不然会导致错误 
			//即只要是height[i]>=mid，都不需要清空size，size可以继续增加下去，看来以后else不能乱用啊 
			if(vis[m[sa[i]]]==0&&m[sa[i-1]]!=m[sa[i]]) {
				vis[m[sa[i]]]=1;
				size++;
			}
		}
		else{
			size=0;
			memset(vis,0,sizeof(vis));
			if(m[sa[i]]>=0) {
				vis[m[sa[i]]]=1;
				size++;
			}
		}
		if(size>n/2) {
			return 1;
		}
	}
	return 0;
}
int main(){
	#ifndef ONLINE_JUDGE
	freopen("in.txt","r",stdin);
	#endif
	int n,num=0,min=-2;
	memset(m,-1,sizeof(m));
	while(scanf("%d",&n),n){
		if(num++) printf("\n");
		int l=0,max=129,min=-1,i,j;
		for(i=0;i<n;i++) scanf("%s",s[i]);
		if(n==1) {
			printf("%s\n",s[0]);
			continue;
		}
		for(i=0;i<n;i++){
			int len=strlen(s[i]);
			for(j=0;j<len;j++) {
				m[l]=i;
				r[l++]=s[i][j];
			}
			m[l]=min--;
			r[l++]=max++;
		}
		r[l]=0;
		da(r,sa,l+1,max);
		calheight(r,sa,l);
		
		int low=0,high=1005,mid;
		while(low+1<high){
			mid=(low+high)>>1;
			if(judge(mid,l,n)) low=mid;
			else high=mid;
		}
		if(low==0) {
			printf("?\n");
			continue;
		}
		else{
			
			int size=0;
			memset(vis,0,sizeof(vis));
			for(i=1;i<=l;i++){
				if(height[i]>=low){
					if(0==vis[m[sa[i]]]&&m[sa[i-1]]!=m[sa[i]]) {
						vis[m[sa[i]]]=1;
						size++;
					}
				}
				else{
					size=0;
					memset(vis,0,sizeof(vis));
					if(m[sa[i]]>=0) {
						vis[m[sa[i]]]=1;
						size++;
					}
				}
				if(size>n/2) {
					for(j=sa[i];j<sa[i]+low;j++){
						printf("%c",r[j]);
					}
					printf("\n");
					while(height[i]>=low&&i<=l) i++;
					i--;
				}
			}
		}
		
	}
	return 0;
}

POJ 3294 Life Forms

标签：后缀数组

原文地址：http://blog.csdn.net/lj94093/article/details/45117055