现在的位置: 首页 > 综合 > 正文

Poj 3581 Sequence (离散化+后缀数组)

2014年02月28日 ⁄ 综合 ⁄ 共 2208字 ⁄ 字号 评论关闭

题意:给定一个数列,将其分为非空的三段。然后分别将三段的数字翻转连接在一起组成新的数列。输出处理后字典序最小的结果。

思路:首先注意必须写成单Case的形式!!!while(scanf("%d",&n)!=EOF)会RE或者WA。。。

因为要翻转,所以读入时直接反着读。由于没有告诉数值的上限,所以接下来离散化。

对于第一段:第一段翻转之后必须最小,因此对翻转过的数列求sa,则排名在前面且sa[i]>1(因为另两段不能为空)的第一个i为第一段。

接下来处理后两段:显然要让第二段的开头尽可能小,但是切去第二段后剩下的部分会直接接在第二段后面,所以不能直接求sa。为保证第三段接在第二段后面时整体最小,所以需要把剩余的串复制到剩余串的后面,再求sa.注意数组要开2倍。

举例见:http://blog.163.com/just_gogo/blog/static/1914390652011823103842787/

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))

const int N = 400000+20;

int cmp(int *r,int a,int b,int l){
    return (r[a]==r[b]) && (r[a+l]==r[b+l]);
}
int wa[N],wb[N],ws[N],wv[N];
int rank[N],height[N];

void DA(int *r,int *sa,int n,int m){
    int i,j,p,*x=wa,*y=wb,*t;
    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*=2,m=p)
    {
        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];
        for(t=x,x=y,y=t,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++;
        //printf("p = %d\n", p );
    }
}
void calheight(int *r,int *sa,int n){
  //  memset(height,0,sizeof(height));
  //  memset(rank,0,sizeof(rank));
    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++);
}


int data[N],sa[N],s[N>>1],tmp[N>>1],n,m;
struct Dis  
{  
    int v;  //存储原数据  
    int id;
    bool operator < (const Dis& b) const  
    {  
        return v<b.v;
    }  
}dis[N>>1];  //离散化数组  


void print (int L,int R)
{
	for (int i=L;i<=R;i++)
		printf("%d\n",s[i]);
}

void Deal ()
{
	DA(data,sa,n+1,m);
	int i,j;
	for (i=1;i<=n;i++)
		if (sa[i]>1)
			break;
	print(sa[i],n-1);
	for (j=0;j<sa[i];j++)
		data[j+sa[i]]=data[j];
	n=sa[i]*2;
	data[n]=0;
	DA(data,sa,n+1,m);
	for (i=1;i<=n;i++)
		if (sa[i]<n/2 && sa[i]>0)
			break;
	print(sa[i],n/2-1);
	print(0,sa[i]-1);
}

int main ()
{  
#ifdef ONLINE_JUDGE
#else
	freopen("read.txt","r",stdin);
#endif
	scanf("%d",&n);
	int i;
	m=n+5;   //m为最大值
	for (i=n-1;i>=0;i--)
	{
		scanf("%d",&s[i]);
		dis[i].v=s[i];
		dis[i].id=i;
	}
	sort (dis,dis+n);
	tmp[0]=1;
	for (i=1;i<n;i++)
		tmp[i]=(dis[i].v==dis[i-1].v?tmp[i-1]:i+1);
	for (i=0;i<n;i++)
		data[dis[i].id]=tmp[i];
	data[n]=0;
	Deal ();
	return 0;
}
/*
8
5 0 3 1 2 3 1 4

Out
0
5
1
3
2
1
3
4
*/

抱歉!评论已关闭.