hdu 5592 ZYB's Premutation (線段樹)

ZYB's Premutation

 

 Accepts: 218

 

 Submissions: 983

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

 

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

問題描述

$ZYB$有一個排列$P$,但他只記得$P$中每個前綴區間的逆序對數,現在他要求你還原這個排列.

$(i,j)(i<j)$被稱爲一對逆序對當且僅當A_i>A_jA​i​​>A​j​​

輸入描述

第一行一個整數$T$表示數據組數。

接下來每組數據：

第一行一個正整數$N$,描述排列的長度.

第二行$N$個正整數A_iA​i​​,描述前綴區間$[1,i]$的逆序對數.

數據保證合法.

$1\le T\le 5$,$1\le N\le 50000$

輸出描述

$T$行每行$N$個整數表示答案的排列.

輸入樣例

1
3
0 1 2

輸出樣例

3 1 2

思路：
稍微轉換一下a[i] - a[i - 1]，就變成了經典的線段樹的題目了

/*======================================================
# Author: whai
# Last modified: 2015-12-07 12:38
# Filename: c.cpp
======================================================*/
#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <stack>

using namespace std;

#define LL __int64
#define PB push_back
#define P pair<int, int>
#define X first
#define Y second

#define LS (2 * n)
#define RS (2 * n + 1)

const int N = 5 * 1e4 + 5;

int a[N];
int ans[N];

struct Seg {
	int l, r, size;
}seg[4 * N];

void up(int n) {
	seg[n].size = seg[LS].size + seg[RS].size;
}

void build(int l, int r, int n) {
	seg[n].l = l; seg[n].r = r;
	if(l == r) {
		seg[n].size = 1;
		return ;
	}
	int mid = (l + r) >> 1;
	build(l, mid, LS);
	build(mid + 1, r, RS);
	up(n);
}

int get_id(int rank, int n) {
	if(seg[n].l == seg[n].r) return seg[n].l;
	if(rank <= seg[LS].size) {
		return get_id(rank, LS);
	} else {
		rank -= seg[LS].size;
		return get_id(rank, RS);
	}
}



void update(int p, int n) {
	if(seg[n].l == seg[n].r) {
		--seg[n].size;
		return ;
	}
	int mid = (seg[n].l + seg[n].r) >> 1;
	if(p <= mid) update(p, LS);
	else update(p, RS);
	up(n);
}

void gao(int n) {
	for(int i = n - 1; i > 0; --i) {
		a[i] -= a[i - 1];
	}
	for(int i = 0; i < n; ++i) {
		a[i] = i + 1 - a[i];
	}
	//for(int i = 0; i < n; ++i) {
	//	cout<<a[i]<<' ';
	//}
	//cout<<endl;
	
	build(1, n, 1);

	for(int i = n - 1; i >= 0; --i) {
		ans[i] = get_id(a[i], 1);
		update(ans[i], 1);
	}

	for(int i = 0; i < n; ++i) {
		printf("%d%c", ans[i], i == n - 1 ? '\n' : ' ');
	}
}

int main() {
	int T;
	scanf("%d", &T);
	while(T--) {
		int n;
		scanf("%d", &n);
		for(int i = 0; i < n; ++i) {
			scanf("%d", &a[i]);
		}
		gao(n);
	}
	return 0;
}