CCF---201709-5---除法---C++---树状数组 or 线段树

试题编号：	201709-5
试题名称：	除法
时间限制：	10.0s
内存限制：	256.0MB

问题描述

	小葱喜欢除法，所以他给了你N个数a1, a2, ⋯, aN，并且希望你执行M次操作，每次操作可能有以下两种：
　　给你三个数l, r, v，你需要将al, al+1, ⋯, ar之间所有v的倍数除以v。
　　给你两个数l, r，你需要回答al + al+1 + ⋯ + ar的值是多少。
输入格式
　　第一行两个整数N, M，代表数的个数和操作的次数。
　　接下来一行N个整数，代表N个数一开始的值。
　　接下来M行，每行代表依次操作。每行开始有一个整数opt。如果opt=1，那么接下来有三个数l, r, v，代表这次操作需要将第l个数到第r个数中v的倍数除以v；如果opt = 2，那么接下来有两个数l, r，代表你需要回答第l个数到第r个数的和。
输出格式
　　对于每一次的第二种操作，输出一行代表这次操作所询问的值。
样例输入
5 3
1 2 3 4 5
2 1 5
1 1 3 2
2 1 5
样例输出
15
14
评测用例规模与约定
　　对于30%的评测用例，1 ≤ N, M ≤ 1000；
　　对于另外20%的评测用例，第一种操作中一定有l = r；
　　对于另外20%的评测用例，第一种操作中一定有l = 1 , r = N；
　　对于100%的评测用例，1 ≤ N, M ≤ 10^5，0 ≤ a1, a2, ⋯, aN ≤ 10^6, 1 ≤ v ≤ 10^6, 1 ≤ l ≤ r ≤ N。

有话说

大规模的区间求和，无非是线段树和树状数组
这题目的坑在于很有可能出现减去数为 0 的情况

1. 被除数为 1 时
2. 被除数小于除数时

对这种两种情况进行特殊处理，第一类直接不处理， 第二类进行筛选处理

上面的是线段树的，下面的是树状数组。可以看到树状数组不但简单而且效率和点更新优化过的线段树相差无几，消耗空间很少

实现代码(树状数组)

#include<iostream>

using namespace std;

typedef long long LL;
const int maxn = 1e5 + 5;
LL tree[maxn];
int a[maxn], n, m;

void add(int i, int num) { // 更新点
	while (i <= n) {
		tree[i] += num;
		i += i & (-i);
	}
}

LL GetSum(int i) { // 获取总值
	LL sum = 0;
	while (i > 0) {
		sum += tree[i];
		i -= i & (-i);
	}
	return sum;
}

int main() {
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	cin >> n >> m;
	for (int i = 1; i <= n; i++) cin >> a[i], add(i, a[i]);
	while (m--) {
		int opt, l, r, v;
		cin >> opt >> l >> r;
		if (opt == 2) cout << GetSum(r) - GetSum(l - 1) << endl;
		else {
			cin >> v;
			if (v == 1) continue;
			for (int i = l; i <= r; i++) {
				if (a[i] >= v && a[i] % v == 0) { // 避免出现num = 0的情况浪费时间
					LL num = a[i] / v;
					add(i, -(a[i] - num));
					a[i] = num;
				}
			}
		}
	}
	return 0;
}

实现代码(线段树)

#include<iostream>
#include<cstring>

using namespace std;

typedef long long LL;
const int maxn = 1e5 + 10;
int n, a[maxn], maxpoint = 1;
LL tree[maxn << 2];;

int lson(int index) {
	return (index << 1) + 1;
}

int rson(int index) {
	return (index + 1) << 1;
}

void update(int index, LL delnum) { //更新:规律自寻
	index += maxpoint - 1; 
	tree[index] -= delnum;
	while (index) {
		index = (index - 1) >> 1;
		tree[index] -= delnum;
	}
}

void build() { // 建树
	memset(tree, 0, sizeof(tree));
	while (maxpoint < n) maxpoint <<= 1;
	for (int i = 0; i < n; i++) {
		update(i, -a[i]);
	}
}

LL query(int index, int now_l, int now_r, int l, int r) {
	if (l == r || (now_r <= l || now_l >= r)) return 0;
	if (l <= now_l && now_r <= r) return tree[index];
	int mid = (now_l + now_r) >> 1;
	return query(lson(index), now_l, mid, l, r) + query(rson(index), mid, now_r, l, r);
}

int main() {
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int m, opt, l, r, v;
	cin >> n >> m;
	for (int i = 0; i < n; i++) cin >> a[i];
	build();
	while (m--) {
		cin >> opt >> l >> r;
		if (opt == 2) cout << query(0, 0, maxpoint, l - 1, r) << endl;
		else {
			cin >> v;
			if (v == 1) continue;
			for (int i = l - 1; i < r; i++) {
				if (a[i] >= v && a[i] % v == 0) {
					LL num = a[i] / v;
					update(i, a[i] - num);
					a[i] = num;
				}
			}
		}
	}
	return 0;
}