2017年第0届浙江工业大学之江学院程序设计竞赛决赛—J（数状数组）

Problem J: qwb又偷懒了

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 799   Solved: 145
[ Submit][ Status][ Web Board]

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"。（不带引号）
每个测试例之后输出一个空行。

Sample Input

3
0 1
0 3
1 1 3
2
0 1
1 2 2

Sample Output

2

zhizhiwuwu

想法：

数状数组

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
long long people[2000000];
long long val[2000000];
int lowbit(int x)
{
    return x&-x;
}


void update(long long *tree, int pos, long long val)
{
    while(pos<2000000)
    {
        tree[pos]+=val;
        pos+=lowbit(pos);
    }
}
long long query(long long *tree, int pos)
{
    long long sum=0;
    while(pos)
    {
        sum+=tree[pos];
        pos-=lowbit(pos);
    }
    return sum;
}
int main()
{
    int pp,t;
    long long sum,x,y,v;
    while(~scanf("%d",&pp))
    {
        memset(people,0,sizeof(people));
        memset(val,0,sizeof(val));
        while(pp--)
        {
            scanf("%d",&t);
            if(t)
            {
                scanf("%lld%lld",&x,&y);
                y++;
                sum=query(people,y)-query(people,x);
                if(sum==0)
                    printf("zhizhiwuwu\n");
                else
                {
                    v=query(val,y)-query(val,x);
                    printf("%lld\n",v/sum);
                }
            }
            else
            {
                scanf("%lld",&x);
                update(people,x+1,1);
                update(val,x+1,x);
            }
        }
        printf("\n");
    }
    return 0;
}