ZOJ-3279 Ants 树状数组 + 二分

题目链接：

https://cn.vjudge.net/problem/ZOJ-3279

题目大意：

有1到n 那个level 每一个level有a[i]只蚂蚁
两种操作

p a b 把第a个level的蚂蚁数量改成b

q  a  查询第a只蚂蚁在哪个level里。

解题思路：

用树状数组动态维护前缀和，二分查找第a个蚂蚁即可

要用printf，不然会超时

 #include<bits/stdc++.h>
 using namespace std;
 const int maxn = 1e5 + ;
 int tree[maxn], n, a[maxn];
 int lowbit(int x)
 {
     return (x & (-x));
 }
 void add(int x, int d)//将x下标的值加上d
 {
     while(x <= n)
     {
         tree[x] += d;
         x += lowbit(x);
     }
 }
 int sum(int x)
 {
     int tot = ;
     while(x)
     {
         tot += tree[x];
         x -= lowbit(x);
     }
     return tot;
 }

 int main()
 {
     while(scanf("%d", &n) != EOF)
     {
         memset(a, , sizeof(a));
         memset(tree, , sizeof(tree));
         for(int i = ; i <= n; i++)
         {
             scanf("%d", &a[i]);
             add(i, a[i]);
         }
         int m, x, y;
         scanf("%d", &m);
         char s[];
         while(m--)
         {
             scanf("%s", s);
             if(s[] == 'p')
             {
                 scanf("%d%d", &x, &y);
                 add(x, -a[x]);//单点修改
                 add(x, y);
                 a[x] = y;
             }
             else if(s[] == 'q')
             {
                 scanf("%d", &x);
                 int l = , r = n, ans;
                 while(l <= r)
                 {
                     int m = (l + r) / ;
                     if(sum(m) >= x)ans = m, r = m - ;
                     else l = m + ;
                 }
                 printf("%d\n", ans);
             }
         }
     }
     return ;
 }