Zoj 3635 <树状数组+二分>

题意：

给出n表示geeks人数..也是位置总数..

给出a1 a2 .. ai .. an 表示第 i 个人想要坐从左往右第 ai 个空位..

给出m表示询问的个数..

给出m个数b1 b2 .. bi .. bn 表示询问第 bi 个人的位置..

思路：

如果模拟过程..

应该是先按要求排好位置..然后输出位置值..

排好位置的过程就用到了二分..

用树状数组表示当前位置前有多少个空位..

然后二分找出满足当前位置前面空位有 ai 个的位置..

Tips：

记得安排好第i个geeks的位置后要modify树状数组..

Code：

 

#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std;
#define clr(x) memset(x, 0, sizeof(x))

int n;       
int c[100001]; 
int lowbit(int x)
{
   return (x)&(-x);
}
void modify(int pos,int x)
{
    while(pos<=n)
    {
        c[pos]+=x;
        pos+=lowbit(pos);
    }
}
int getSum(int x)
{
      int sum=0;
      while(x>0)
      {
         sum+=c[x];
         x-=lowbit(x);
      }
      return sum;
}

int ans[50010];
int arr[50010];
int main()
{
    int i, j, k;
    int m, tmp;
    while(scanf("%d", &n) != EOF)
    {
        clr(c);
        clr(ans);
        clr(arr);
        for(i = 1; i <= n; ++i)
        modify(i, 1);

        for(i = 1; i <= n; ++i){
            scanf("%d", &tmp);
            int r = n, l = 1, mid;
            while(r > l)
            {
                mid = (r+l)/2;
                if(getSum(mid) >= tmp) r = mid;
                else l = mid+1;
            }
            
            arr[i] = r;
            modify(r, -1);
        }


        scanf("%d", &m);
        for(i = 0; i < m; ++i){
            scanf("%d", &tmp);
            ans[i] = arr[tmp];
        }

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

    return 0;
}