2017年第0届浙江工业大学之江学院程序设计竞赛决赛—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"。（不带引号）
每个测试例之后输出一个空行。

Sample Input

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

Sample Output

2

zhizhiwuwu

思路：简单题，用两个树状数组，一个维护收入，一个维护人数。

需要注意的点是：会有0收入的人，因为线段树和树状数组只能从1开始进行建二叉树，所以把整体右移一个位置即可。

Code:

#include <bits/stdc++.h>
#define LL long long
using namespace std;
const int n = 1000001;
LL c1[n+5], c2[n+5];
int lowbit(int x)
{
	return x&-x;
}
void update(LL *c, int k, LL t)
{
	while(k <= n)
	{
		c[k] += t;
		k += lowbit(k);
	}
}
LL query(LL *c, int k)
{
	LL ans = 0;
	while(k)
	{
		ans += c[k];
		k -= lowbit(k);
	}
	return ans;
}
int main()
{
	int t, a;
	LL b, p;
	while(~scanf("%d", &t))
	{
		memset(c1, 0, sizeof c1);
		memset(c2, 0, sizeof c2);
		for(int i = 1; i <= t; ++i)
		{
			scanf("%d", &a);
			if(a == 0) 
			{
				scanf("%lld", &b);
				update(c1, b+1, b);
				update(c2, b+1, 1);
			}
			else
			{
				scanf("%lld %lld", &b, &p);
				LL k = (query(c2, p+1) - query(c2, b));
				if(k == 0) printf("zhizhiwuwu\n");
				else
				{
					LL ans = query(c1, p+1) - query(c1, b);
					if(ans/k) printf("%lld\n", ans/k);
					else printf("zhizhiwuwu\n");
				}
			}
		}
		puts("");
	}
	return 0;
}

继续加油~