【BestCoder Round 65C】【树状数组动态查找第k大 O(nlogn)】ZYB's Premutation 告诉你前i个数中的逆序对数让你还原全排列

本文主要是介绍【BestCoder Round 65C】【树状数组动态查找第k大 O(nlogn)】ZYB's Premutation 告诉你前i个数中的逆序对数让你还原全排列，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

ZYB's Premutation

Accepts: 218

Submissions: 983

Time Limit: 2000/1000 MS (Java/Others)

Memory Limit: 131072/131072 K (Java/Others)

问题描述

 $Z Y B$ 有一个排列 $P$ ,但他只记得 $P$ 中每个前缀区间的逆序对数,现在他要求你还原这个排列. $(i, j) (i < j)$ 被称为一对逆序对当且仅当 $A_i>A_j$

输入描述

第一行一个整数 $T$ 表示数据组数。接下来每组数据：第一行一个正整数 $N$ ,描述排列的长度.第二行 $N$ 个正整数 $A_i$ ,描述前缀区间 $[1, i]$ 的逆序对数.数据保证合法. $\leq T \leq 5$ , $\leq N \leq 50000$

输出描述

 $T$ 行每行 $N$ 个整数表示答案的排列.

输入样例

1
3
0 1 2

输出样例

3 1 2

【BestCoder Round 65C】【树状数组动态查找第k大 O(n(logn)^2)】ZYB's Premutation 告诉你前i个数中的逆序对数让你还原全排列

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<ctype.h>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<bitset>
#include<algorithm>
#include<time.h>
using namespace std;
void fre(){freopen("c://test//input.in","r",stdin);freopen("c://test//output.out","w",stdout);}
#define MS(x,y) memset(x,y,sizeof(x))
#define MC(x,y) memcpy(x,y,sizeof(x))
#define MP(x,y) make_pair(x,y)
#define ls o<<1
#define rs o<<1|1
typedef long long LL;
typedef unsigned long long UL;
typedef unsigned int UI;
template <class T1,class T2>inline void gmax(T1 &a,T2 b){if(b>a)a=b;}
template <class T1,class T2>inline void gmin(T1 &a,T2 b){if(b<a)a=b;}
const int N=5e4+10,M=0,Z=1e9+7,ms63=1061109567;
int casenum,casei;
int a[N];
int b[N];
int ans[N];
int n;
void add(int x,int v)
{for(;x<=n;x+=x&-x)b[x]+=v;
}
int cnt(int x)
{int tmp=0;for(;x;x-=x&-x)tmp+=b[x];return tmp;
}
int main()
{scanf("%d",&casenum);for(casei=1;casei<=casenum;++casei){MS(b,0);scanf("%d",&n);for(int i=1;i<=n;++i){scanf("%d",&a[i]);//记录有多少个数比这个数大add(i,1);}for(int i=n;i>=1;--i){int kth=i-(a[i]-a[i-1]);int l=1;int r=n;while(l<r){int m=(l+r)>>1;if(cnt(m)>=kth)r=m;else l=m+1;}ans[i]=l;add(l,-1);}for(int i=1;i<n;++i)printf("%d ",ans[i]);printf("%d\n",ans[n]);}return 0;
}
/*
【题意】
告诉你一个1~n（1<=n<=50000）的全排列。
然后，告诉你前i个数中总共有多少个逆序对。
让你还原这个全排列。【类型】
树状数组+二分【分析】
告诉你前i个数共有几个逆序对，事实上就告诉了你，第i个数和之前的数共计形成了多少个逆序对。
也就是说，我们就知道了，对于第i个数，它在前i个数中排第几。
然而，我们发现这个很多时候并没有什么卵用，我们还是确定不了数值。除了——最后一个数。最后一个数在所有数中排第几，我们就能知道这个数是几。
然后，因为后面所有数都不涉及到这最后一个数的信息，于是我们把最后一个数从我们当前数的集合中抹去。
接下来，用同样的方法处理倒数第二个数，倒数第三个数，直到最后处理完所有的n个数。我们发现，为了实现刚才的想法，我们需要知道动态知道当前第几大的数是几。
这个我们可以通过线段树，用O（nlogn）实现。
也可以通过树状数组+二分，用O（n(logn)^2）实现。
就是二分最后一个数是m，看看前m个数中数的个数是否达到kth。
树状数组虽然多了一个logn，但是常数很小，所以一样可以快速AC。然而，接下来还要为大家介绍一种O(nlogn)的树状数组做法。~~【时间复杂度&&优化】
O（n(logn)^2） ->O（nlogn）【数据】
0 1 2 3 4
0 1 3 6 10*/

【BestCoder Round 65C】【树状数组动态查找第k大 O(nlogn)】ZYB's Premutation 告诉你前i个数中的逆序对数让你还原全排列

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<ctype.h>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<bitset>
#include<algorithm>
#include<time.h>
using namespace std;
void fre(){freopen("c://test//input.in","r",stdin);freopen("c://test//output.out","w",stdout);}
#define MS(x,y) memset(x,y,sizeof(x))
#define MC(x,y) memcpy(x,y,sizeof(x))
#define MP(x,y) make_pair(x,y)
#define ls o<<1
#define rs o<<1|1
typedef long long LL;
typedef unsigned long long UL;
typedef unsigned int UI;
template <class T1,class T2>inline void gmax(T1 &a,T2 b){if(b>a)a=b;}
template <class T1,class T2>inline void gmin(T1 &a,T2 b){if(b<a)a=b;}
const int N=5e4+10,M=0,Z=1e9+7,ms63=1061109567;
int casenum,casei;
int a[N];
int b[N];
int ans[N];
int n;
void add(int x,int v)
{for(;x<=n;x+=x&-x)b[x]+=v;
}
int cnt(int x)
{int tmp=0;for(;x;x-=x&-x)tmp+=b[x];return tmp;
}
int main()
{scanf("%d",&casenum);for(casei=1;casei<=casenum;++casei){MS(b,0);scanf("%d",&n);for(int i=1;i<=n;++i){scanf("%d",&a[i]);//记录有多少个数比这个数大add(i,1);}int top;for(top=1;top<=n;top<<=1);top>>=1;for(int i=n;i>=1;--i){int kth=i-(a[i]-a[i-1]);int p=0;int sum=0;for(int j=top;j;j>>=1)if(p+j<=n&&sum+b[p+j]<kth){p+=j;sum+=b[p];}++p;ans[i]=p;add(p,-1);}for(int i=1;i<n;++i)printf("%d ",ans[i]);printf("%d\n",ans[n]);}return 0;
}
/*
【题意】
告诉你一个1~n（1<=n<=50000）的全排列。
然后，告诉你前i个数中总共有多少个逆序对。
让你还原这个全排列。【类型】
树状数组+二分【分析】
告诉你前i个数共有几个逆序对，事实上就告诉了你，第i个数和之前的数共计形成了多少个逆序对。
也就是说，我们就知道了，对于第i个数，它在前i个数中排第几。
然而，我们发现这个很多时候并没有什么卵用，我们还是确定不了数值。除了——最后一个数。最后一个数在所有数中排第几，我们就能知道这个数是几。
然后，因为后面所有数都不涉及到这最后一个数的信息，于是我们把最后一个数从我们当前数的集合中抹去。
接下来，用同样的方法处理倒数第二个数，倒数第三个数，直到最后处理完所有的n个数。我们发现，为了实现刚才的想法，我们需要知道动态知道当前第几大的数是几。
这个我们可以通过线段树，用O（nlogn）实现。
也可以通过树状数组+二分，用O（n(logn)^2）实现。
就是二分最后一个数是m，看看前m个数中数的个数是否达到kth。
树状数组虽然多了一个logn，但是常数很小，所以一样可以快速AC。然而，接下来还要为大家介绍一种O(nlogn)的树状数组做法。~~
具体是怎样呢？
先回顾一下树状数组，以 +=x&-x 向上加权， -=x&-x 向下统计的做法为例。
我们会统计前1，2，4，8，16个数的权值到相应的计数桶里。
所以，如果我们要求前k大的，类似于倍增的做法，我们就跳着找——
前1个数内数的个数是否有k-1个
前2个数内数的个数是否有k-1个
前4个数内数的个数是否有k-1个
前8个数内数的个数是否有k-1个
如果没有，我们就计数+，而且把下标移过去，同时降低我们的倍增幅度（因为显然不会再倍增相同幅度）
如果有，我们就不计数+，不把下标移过去，但依然要降低我们的倍增幅度（因为我们这个幅度不能延伸，所以要看更小的范围）
直到我们得到最大的延展范围，使得范围内的数的个数恰好为k-1个。
于是，我们使得的下标向后移1，范围内数的个数就会恰好是k个。充分利用了树状数组的性质，使得做法是O（nlogn）【时间复杂度&&优化】
O（n(logn)^2） ->O（nlogn）【数据】
0 1 2 3 4
0 1 3 6 10*/

这篇关于【BestCoder Round 65C】【树状数组动态查找第k大 O(nlogn)】ZYB's Premutation 告诉你前i个数中的逆序对数让你还原全排列的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！