[POJ2182]Lost Cows（树状数组，二分）

题目链接：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 = 8080;
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 = 0;
  while(i) {
    ret += bit[i];
    i -= lowbit(i);
  }
  return ret;
}

int ub(int val) {
  int lo = 1, hi = n;
  while(lo <= hi) {
    int mid = (lo + hi) >> 1;
    if(sum(mid) >= val) hi = mid - 1;
    else lo = mid + 1;
  }
  return lo;
}

void dfs(int i) {
  if(i == 0) return;
  int pos = ub(a[i]+1);
  update(pos, -1);
  dfs(i-1);
  printf("%d
", pos);
}

int main() {
  //freopen("in", "r", stdin);
  while(~scanf("%d", &n)) {
    memset(a, 0, sizeof(a));
    memset(bit, 0, sizeof(bit));
    for(int i = 1; i <= n; i++) update(i, 1);
    for(int i = 2; i <= n; i++) scanf("%d", &a[i]);
    dfs(n);
  }
  return 0;
}