题目:
http://poj.org/problem?id=3581
题意:
给定一个数列,分成三段,每段分别反转,求能得到的字典序最小的序列是什么,要求每段非空。
分析:
第一段容易求得,就是把原序列反转后,求一遍后缀数组,取字典序最小的并且满足三段非空的那段序列。
第二三段,是把原序列除去第一段后反转,再复制成双倍长度,求后缀数组,取字典序最小且满足此两段非空的序列。
注意:
需用scanf print,cin cout会超时。
代码:
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int Tmax=200005;
int n,data[Tmax],rev[Tmax*2],sa[Tmax*2],rank[Tmax*2],tmp[Tmax*2];
int k,len;
bool cmp(int i,int j)
{
if(rank[i]!=rank[j]) return rank[i]<rank[j];
int ri=i+k<=len?rank[i+k]:-1;
int rj=j+k<=len?rank[j+k]:-1;
return ri<rj;
}
void suffix(int *A)
{
for(int i=0;i<=len;i++)
{
sa[i]=i;
rank[i]=i<len?A[i]:-1;
}
for(k=1;k<=len;k*=2)
{
sort(sa,sa+len+1,cmp);
tmp[sa[0]]=0;
for(int i=1;i<=n;i++)
{
tmp[sa[i]]=tmp[sa[i-1]];
if(cmp(sa[i-1],sa[i])==true) tmp[sa[i]]++;
}
for(int i=0;i<=n;i++)
rank[i]=tmp[i];
}
return;
}
int main()
{
int p1,m;
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&data[i]);
reverse_copy(data,data+n,rev);
len=n;
suffix(rev);
for(int i=0;i<=n;i++)
{
p1=n-sa[i]-1;
m=n-p1-1;
if(p1>=0&&m>=2) break;
}
for(int i=p1;i>=0;i--)
{
printf("%d\n",data[i]);
}
reverse_copy(data+p1+1,data+n,rev);
reverse_copy(data+p1+1,data+n,rev+m);
len=m*2;
suffix(rev);
for(int i=0;i<=m*2;i++)
{
if(sa[i]!=0&&sa[i]<m)
{
for(int j=0;j<m;j++)
printf("%d\n",rev[sa[i]+j]);
break;
}
}
return 0;
}
请允许蒟蒻庆祝一波