题目链接:http://poj.org/problem?id=2182
题意:给定1~n个数和n个位置,已知ai表示第i个位置前有ai个数比当前位置的数小,求这个排列。
和刚才YY的题意蛮接近的,用树状数组维护当前数组内数字分别是第几大的,倒着查询,每次查尽可能靠右的位置,可以保证取的数字是未取到并且不会冲突。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <cstdlib>
using namespace std; #define lowbit(x) x & (-x)
const int maxn = ;
int n;
int a[maxn];
int bit[maxn]; void update(int i, int x) {
while(i <= n) {
bit[i] += x;
i += lowbit(i);
}
} int sum(int i) {
int ret = ;
while(i) {
ret += bit[i];
i -= lowbit(i);
}
return ret;
} int ub(int val) {
int lo = , hi = n;
while(lo <= hi) {
int mid = (lo + hi) >> ;
if(sum(mid) >= val) hi = mid - ;
else lo = mid + ;
}
return lo;
} void dfs(int i) {
if(i == ) return;
int pos = ub(a[i]+);
update(pos, -);
dfs(i-);
printf("%d\n", pos);
} int main() {
//freopen("in", "r", stdin);
while(~scanf("%d", &n)) {
memset(a, , sizeof(a));
memset(bit, , sizeof(bit));
for(int i = ; i <= n; i++) update(i, );
for(int i = ; i <= n; i++) scanf("%d", &a[i]);
dfs(n);
}
return ;
}