[Codeforces 1208D]Restore Permutation (树状数组)

[Codeforces 1208D]Restore Permutation (树状数组)

题面

有一个长度为n的排列a。对于每个元素i，\(s_i\)表示\(\sum_{j=1,a_j<a_i}^i a_j\).即前面比它小的元素的值之和。

给出\(s_1,s_2 \dots s_n\),求a

分析

考虑如何求\(a_n\),\(s_n\)实际上表示的是1~n中比\(a_n\)小的所有数的和，可以直接求出\(a_n\)

然后我们可以倒序求\(a_i\)，求到\(a_i\)的时候，我们已经知道\(a_i\)前面有哪些数，只是不知道顺序。维护一个数据结构存储这些数的集合，并存储从小到大排列后的前缀和，然后二分出最后一个前缀和\(\leq s_i\)的位置。

比如最后一个样例：

input:
5
0 1 1 1 10

output:
1 4 3 2 5

我们求到第4个数的时候，由于第5个数已经求出来是5，那前面的数就是{1,2,3,4}，2之前的前缀和是1，1之前的前缀和是0，那么二分出的点就是2.

具体实现可以用树状数组，初始时候给第i个位置加i，求出某个数的值为x之后把第x个位置-x,相当于把x移出集合

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#define maxn 200000
using namespace std;
typedef long long ll;
int n;
struct fenwick_tree{
	ll c[maxn+5];
	inline int lowbit(int x){
		return x&(-x);
	}
	inline void update(int x,int v){
		for(int i=x;i<=n;i+=lowbit(i)){
			c[i]+=v;
		}
	}
	inline ll query(int x){
		ll ans=0;
		for(int i=x;i>0;i-=lowbit(i)) ans+=c[i];
		return ans;
	}
}T;
ll a[maxn+5];
int ans[maxn+5];
int bin_search(int l,int r,ll sum){
	int ans=n;
	int mid;
	while(l<=r){
		mid=(l+r)>>1;
		if(T.query(mid-1)<=sum){
			ans=mid;
			l=mid+1;
		}else r=mid-1;
	}
	return ans;
}

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%I64d",&a[i]);
	}
	for(int i=1;i<=n;i++) T.update(i,i);
	for(int i=n;i>=1;i--){
		int k=bin_search(1,n,a[i]);
		ans[i]=k;
		T.update(k,-k);
	}
	for(int i=1;i<=n;i++) printf("%d ",ans[i]);
}