[树状数组]校门外的树

[树状数组]校门外的树
题目描述

校门外有很多树，有苹果树，香蕉树，有会扔石头的，有可以吃掉补充体力的……
如今学校决定在某个时刻在某一段种上一种树，保证任一时刻不会出现两段相同种类的树，现有两个操作：
K =1，读入 l , r 表示在 l ~ r 之间种上的一种树
K =2，读入 l , r 表示询问 l ~ r之间能见到多少种树
（ l , r >0）

输入
第一行 n , m 表示道路总长为 n ，共有 m 个操作
接下来 m 行 是 操作

输出
对于每个 k=2 输出一个答案
样例输入
5 4
1 1 3
2 2 5
1 2 4
2 3 5
样例输出
1
2
提示
n,m<=5*(10^4)

这一题相当于种一次树就标记一个区间1，当询问一个区间2时就是计算有多少个区间1至少能有一个元素落在区间2
相当于左括号减去右括号

#include<stdio.h>
#include<algorithm>

const int N = 50005;
int n, m;
int a[N], b[N];
int lowbit(int x)
{
	return (x & (-x));
}

void Update(int x, int add, int* arr)
{
	for (int i = x; i <= n; i += lowbit(i)) {
		arr[i] += add;
	}

}

int Sum(int x, int* arr)
{
	int ans = 0;
	for (int i = x; i > 0; i -= lowbit(i)) {
		ans += arr[i];
	}
	return ans;
}

int main()
{
	scanf("%d %d", &n, &m);
	int k, l, r;
	for (int i = 1; i <= m; i++) {
		scanf("%d %d %d", &k, &l, &r);
		if (k == 1) {
			Update(l, 1, a);
			Update(r + 1, 1, b);
		}
		else {
			printf("%d\n", Sum(r, a) - Sum(l, b));
		}
	}
}

标签：arr,return,树状,int,门外,数组,ans,区间,include
来源： https://www.cnblogs.com/tzp-empty-hya/p/15832101.html