Description
小可可是音乐学院的一名学生,他需要经常创作乐曲完成老师布置的作业。
可是,小可可是一个懒惰的学生。所以,每次完成作业时,他不会重新创作一首新的乐曲,而是去修改上一次创作过的乐曲作为作业交给老师。小可可的乐曲由N个音调不同的音符组成,分别记为音符1…N。因此,他创作的乐曲是由1…N的一个排列构成,例如N=5时,他创作的乐曲可能为:2,1,3,5,4。但是,小可可每一次会按照一定的要求修改上一次创作的乐曲。他规定,修改过后的乐曲必须与上一次创作的乐曲的悦耳值相同。所谓悦耳值就是他所创作的乐曲,也就是1…N的排列中逆序对的个数。逆序对是指对于1…N的一个排列A1,A2,…,An中的两个数Ai,Aj,满足i< j而Ai>Aj,例如:2,1,3,5,4 这个排列中有2个逆序对,分别为:(2,1),(5,4)。可是,满足条件的排列有很多,小可可会选择在这些满足条件的排列中字典序大于上次创作乐曲的排列的字典序,且字典序尽量小的那一个排列作为新的乐曲。这里的字典序指:排列A:A1,A2…An和排列B:B1,B2…Bn,若存在一个数k,使得Ak< Bk,且A1=B1,A2=B2,……,Ak-1=Bk-1,我们就称排列A的字典序小于排列B的字典序。
由于小可可最近要参加学校的篮球比赛,他没有空余时间完成老师布置的作业,于是他希望作为他好友的你帮助他完成作业。
Data Constraint
30%的数据,1≤N≤10;
60%的数据,1≤N≤1000;
100%的数据,1≤N≤500000。
Solution
我们将条件一个个解决。
1、要求满足条件的排列中字典序大于上次创作乐曲的排列的字典序,且字典序尽量小的那一个排列作为新的乐曲。所以这启示我们要从后往前找到第一个满足修改条件的点,使它的权值大于当前点,前面的点都不作改变,这就保证了字典序大于原序列且尽量小。
2、要保证逆序对不变。所以我们要保证开始修改的点之后的点是可实现逆序对等于当前序列的。
所以我们从后往前找到一个位置x满足:1、它后面存在一个比它大的点。2、当x放了一个权值比他大的权值最小点时,剩下的点从小到大放置所造成的逆序对数一定小于原序列的逆序对数。
接下来从这个点x往后做。我们先强制在点x放上一个权值比a[x]大的权值最小点,然后剩下的点我们这样考虑:当前i-1个点已经确定,假设当前i号点放了一个权值第k小的点,剩下的点从大到小倒着放置,那么它造成的逆序对为
k−1+(n−i)∗(n−i−1)2
,我们要使在满足
k−1+(n−i)∗(n−i−1)2+x>=sum(sum表示原序列的逆序对,x表示前i−1号点的逆序对)
k尽量小,k可直接算出,放置在第i位上,往下做i+1位。因为假设满足
k−1+(n−i)∗(n−i−1)2+x>=sum
,那么我一定能够能做到当前i号点放了一个权值第k小的点,剩下的点按某种方式放置实现逆序对等于原序列。如此类推往下做即可。
Code
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const ll maxn=5e5+5;
ll a[maxn],f[maxn],d[maxn],b[maxn],g[maxn*4];
ll n,i,t,j,k,l,x,y,sum,p,q,mid,r;
ll lowbit(ll x){
return x&(-x);
}
void insert(ll x){
if (x>n) return;f[x]++;insert(x+lowbit(x));
}
int find(ll x){
if (x<1) return 0;
return f[x]+find(x-lowbit(x));
}
void insert1(int l,int r,int v,int x){
int mid=(l+r)/2;
if (l==r){
g[v]++;return;
}
if (mid>=x) insert1(l,mid,v*2,x);
else insert1(mid+1,r,v*2+1,x);
g[v]=g[v*2]+g[v*2+1];
}
void find1(int l,int r,int v,int x){
int mid=(l+r)/2;
if (!g[v]) return;
if (l==r){
b[q]=l;g[v]--;return;
}
if (mid>=x) find1(l,mid,v*2,x);
if (!b[q]) find1(mid+1,r,v*2+1,x);
g[v]=g[v*2]+g[v*2+1];
}
void find2(int l,int r,int v,int x,int y){
int mid=(l+r)/2;
if (!g[v]) return;
if (l>=x && r<=y){
k+=g[v];return;
}
if (mid>=x && l<=y) find2(l,mid,v*2,x,y);
if (mid<y && r>=x) find2(mid+1,r,v*2+1,x,y);
}
void find3(int l,int r,int v,int x){
int mid=(l+r)/2;
if (l==r){
b[j]=l;g[v]--;return;
}
if (g[v*2]>=x) find3(l,mid,v*2,x);
else find3(mid+1,r,v*2+1,x-g[v*2]);
g[v]=g[v*2]+g[v*2+1];
}
int main(){
// freopen("data.in","r",stdin);freopen("data.out","w",stdout);
scanf("%lld",&n);
for (i=1;i<=n;i++){
scanf("%lld",&a[i]);
d[i]=a[i]-1-find(a[i]);insert(a[i]);sum+=d[i];
}p=sum;insert1(1,n,1,a[n]);
for (i=n-1;i>=1;i--){
insert1(1,n,1,a[i]);
sum-=d[i];
t=n-i;k=0;find2(1,n,1,1,a[i]);
if (sum+(t+1)*t/2==p || sum+k>p || d[i]==n-i) continue;
q=i;find1(1,n,1,a[i]+1);sum+=k;
for (j=q+1;j<=n;j++){
l=1;
r=n-j+1;t=n-j-1;
while (l<r){
mid=(l+r)/2;
if (sum+mid-1+(t+1)*t/2<p) l=mid+1;
else r=mid;
}
find3(1,n,1,l);sum+=l-1;
}
break;
}
for (i=1;i<q;i++)
printf("%lld ",a[i]);
for(i=q;i<n;i++)
printf("%lld ",b[i]);
printf("%lld\n",b[i]);
}