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

古城微笑少年丶 2021-11-11 01:40 270阅读 0赞

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

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

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

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

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

以下举例转自 [poj3581 Sequence(后缀数组) - Damonbaby的日志 - 网易博客][poj3581 Sequence_ - Damonbaby_ -]

刚开始的时候觉得没有必要将剩下的数字复制一片贴在剩下的数字后面，所以wa了很多次。。。  
最后在discuss里面看见一组数据：  
9  
8 4 -1 5 0 5 0 2 3  
第一步：  
3 2 0 5 0 5 -1 4 8 对应输出 -1 4 8  
第二步  
3 2 0 5 0 5（开始的时候我并没有复制一遍） 对应输出：0 5  
第三步  
3 2 0 5 对应输出： 3 2 0 5  
可以看见这样做是不对的。。  
必须要将剩下的字符串复制一遍贴在后面，然后再来求后缀数组。。。  
正解：  
第一步：  
3 2 0 5 0 5 -1 4 8 对应输出 -1 4 8  
第二步  
3 2 0 5 0 5 3 2 0 5 0 5 对应输出： 0 5 0 5；  
第三步  
3 2 对应输出：3 2；

#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
    */

[while_scanf_d_n_EOF]: http://poj.org/showmessage?message_id=165544
[poj3581 Sequence_ - Damonbaby_ -]: http://blog.163.com/just_gogo/blog/static/1914390652011823103842787/