[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 = ;
 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 ;
 }