之江学院2017ACM 校赛 Problem J: qwb又偷懒了(树状数组)

Description

qwb最近在做一个群众收入统计。ta非常懒，以至于忘记了今天领导要来视察。所以急忙催下属去做统计。
在接下来长度为n的时间里,每个单位时间都有事情发生,可能会发下以下两种事件：

1)下属递交了一份调查报告，由于太匆忙，上面只有一个整数x，代表一个居民的收入。
2)领导来视察了，领导会来询问，收入在区间[l,r]内的居民的平均收入，qwb需要给出回答。

qwb非常讨厌小数，所以qwb上报时都会省略小数部分。如果上报时统计的人数为0，qwb就暴露了他偷懒的事情，他就会zhizhiwuwu。

Input

多组测试数据，处理到文件末尾。

每组测试数据的第一行为一个正整数n(0<=100000),确保所有的n的和不超过300000

接下来n行，
当第一个数为0时，代表操作1，后面跟着一个整数x(0<=x<=1000000)，意义如题目所述。
当第一个数为1时，代表操作2，后面跟着两个整数l，r(0<=l<=r<=1000000)，意义如题目描述。

Output

对于每一个领导的询问，给出一个回答，如果统计区间的人数为零，则输出"zhizhiwuwu"。（不带引号）
每个测试例之后输出一个空行。

思路：用两个树状数组分别维护每个工资的人数和每个工资的总工资。

注意：工资可能为0，会RE，读入的时候先都把工资右移一位。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 5e6+5;
ll treeNum[maxn], treeVal[maxn];
int lowbit(int x) { return x&-x; }
 
void update(ll *tree, int pos, ll val)
{
    while(pos < maxn)
    {
        tree[pos] += val;
        pos += lowbit(pos);
    }
}
 
ll query(ll *tree, int pos)
{
    ll sum = 0;
    while(pos)
    {
        sum += tree[pos];
        pos -= lowbit(pos);
    }
    return sum;
}
 
int main(void)
{
    int q;
    while(~scanf("%d", &q))
    {
        memset(treeNum, 0, sizeof(treeNum));
        memset(treeVal, 0, sizeof(treeVal));
        while(q--)
        {
            int cmd, x, y;
            scanf("%d%d", &cmd, &x);
            x++;
            if(!cmd)
            {
                update(treeNum, x, 1);
                update(treeVal, x, x-1);
            }
            else
            {
                scanf("%d", &y);
                y++;
                ll num = query(treeNum, y)-query(treeNum, x-1);
                ll val = query(treeVal, y)-query(treeVal, x-1);
                if(!num) puts("zhizhiwuwu");
                else
                {
                    ll ans = val/num;
                    printf("%lld\n", ans);
                }
            }
        }
        puts("");
    }
    return 0;
}