乐曲创作

乐曲创作
题目描述
小可可是音乐学院的一名学生，他需要经常创作乐曲完成老师布置的作业。
可是，小可可是一个懒惰的学生。所以，每次完成作业时，他不会重新创作一首新的乐曲，而是去修改上一次创作过的乐曲作为作业交给老师。小可可的乐曲由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的字典序。

由于小可可最近要参加学校的篮球比赛，他没有空余时间完成老师布置的作业，于是他希望作为他好友的你帮助他完成作业。

输入
第1行1个正整数N，表示小可可的乐曲由N个音调不同的音符组成。

第2行为N个不同的正整数，表示1…N的一个排列，即小可可上次创作的乐曲。

输出
输出只有一行，为1…N的一个排列，表示你帮助小可可修改后的乐曲。

数据保证有满足条件的解。

样例输入
3
2 3 1
样例输出
3 1 2
提示
【样例2】

music.in

music.out

5

2 1 3 5 4

2 1 4 3 5

【数据范围】

30%的数据，1≤N≤10；

60%的数据，1≤N≤1000；

100%的数据，1≤N≤500000。

$$先找出前面最多几位不用变$$
$$后面几位用贪心，尽可能放最小的可以满足条件的数$$

#include<iostream>
#include<cstdio>
using namespace std;
#define ll long long
#define lowbit(x) ((x)&(-x))
int a[600009],b[600009],h[600009],ma[600009];
ll c[600009];
int tree[2400009];
int n;
void add(int x,int z){
  while (x<=n){
    h[x]+=z;
    x+=lowbit(x);
  }
}
int sum(int x){
  int ans=0;
  while (x>0){
    ans+=h[x];
    x-=lowbit(x);
  }
  return ans;
}

void change(int p,int l,int r,int y,int z){
  if(l==r){
    tree[p]+=z;
    return;
  }
  int mid=(l+r)/2;
  if(y<=mid)change(p+p,l,mid,y,z);
  else change(p+p+1,mid+1,r,y,z);
  tree[p]=tree[p+p]+tree[p+p+1];
}
int find(int p,int l,int r,int y){
  if(l==r)return l;
  int mid=(l+r)/2;
  if(y<=tree[p+p])return find(p+p,l,mid,y);
  else return find(p+p+1,mid+1,r,y-tree[p+p]);
}
int main(){
  scanf("%d",&n);
  for (int i=1;i<=n;++i) scanf("%d",&a[i]);
  ll num=0;
  for (int i=n;i>=1;--i){
    b[i]=sum(a[i]);
    num+=b[i];
    add(a[i],1);
  }
  for (int i=1;i<=n;++i) c[i]=c[i-1]+b[i];
  ma[n+1]=0;
  for (int i=n;i>=1;--i) ma[i]=max(ma[i+1],a[i]);
  int ans=1;
  for (int i=n;i>=1;--i){
    ll t=n-i+1;
    if(ma[i]>a[i]&&c[i-1]+b[i]+1<=num&&c[i-1]+t*(t-1)/2+i-1>=num){
      ans=i; 
      break;
    }
  }
    //cout<<ans<<endl;
  int tt=1<<29;
  for (int i=ans+1;i<=n;++i)
    if (a[i]>a[ans]) tt=min(tt,a[i]);
  ll QQ=0;
  for (int i=ans;i<=n;++i)
    if (a[i]!=tt) change(1,1,n,a[i],1),QQ+=(a[i]<tt);
  a[ans]=tt;
  QQ+=c[ans-1];
  //cout<<QQ<<endl;
  for (int i=ans+1;i<=n;++i){
    int l=1,r=n-i+1;
    ll t=n-i;
    while (l<r){
      ll mid=(l+r)/2;
      ll k=t*(t-1)/2+QQ+mid-1;
      if (k<num) l=mid+1;
      else r=mid;
    }
    a[i]=find(1,1,n,l);
    QQ+=l-1;
    change(1,1,n,a[i],-1);
  }
  for (int i=1;i<n;++i) printf("%d ",a[i]);
  printf("%d\n",a[n]); 
  //while (1);
  return 0;
}