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 + 10;
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 = 0;
    while(x)
    {
        tot += tree[x];
        x -= lowbit(x);
    }
    return tot;
}

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