一个简单的整数问题2 ——分块思想

分块本质上是一种暴力算法，是将一个N长的数列分成sqrt(N)块。当有区间操作覆盖了完整的一个块时，就可以只对块进行操作，而不是具体操作到点。这样就能节省下许多时间。
给定一个长度为 N 的数列 A，以及 M 条指令，每条指令可能是以下两种之一：

C l r d，表示把 A[l],A[l+1],…,A[r] 都加上 d。
Q l r，表示询问数列中第 l∼r 个数的和。
对于每个询问，输出一个整数表示答案。

输入格式
第一行两个整数 N,M。

第二行 N 个整数 A[i]。

接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。

输出格式
对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围
1≤N,M≤1e5,
|d|≤10000,
|A[i]|≤1e9

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cmath>
using namespace std;
typedef long long ll;
const int N = 100010, M = 350; //M是分块的数量，为根号N;
int n, m, len;
ll add[M], sum[M]; //每段记录add和sum
int w[N];//记录元素

int get(int i) { //映射，i位于哪一块
	return i / len;
}

void change(int l, int  r, int d) {
	if (get(l) == get(r)) { //如果刚好再同一段,直接暴力
		for (int i = l; i <= r; i++) {
			w[i] += d;
			sum[get(i)] += d;
		}
	} else {
		int i = l, j = r;
		while (get(i) == get(l)) { //先暴力做，把整段外的那些"多余"处理掉
			w[i] += d, sum[get(i)] += d, i++;
		}
		while (get(j) == get(r)) {
			w[j] += d, sum[get(j)] += d, j--;
		}
		for (int k = get(i); k <= get(j); k++) { //中间完整段
			sum[k] += len * d, add[k] += d;
		}
	}
}

ll query(int l, int r) {
	ll ans = 0;
	if (get(l) == get(r)) {
		for (int i = l; i <= r; i++) {
			ans += w[i] + add[get(i)];
		}
	} else {
		int i = l, j = r;
		while (get(i) == get(l))
			ans += w[i] + add[get(i)],i++;
		while (get(j) == get(r))
			ans += w[j] + add[get(j)],j--;
		for (int k = get(i); k <= get(j); k++) {
			ans += sum[k];
		}
	}
	return ans;
}

int main() {
	cin >> n >> m;
	len = sqrt(n);
	for (int i = 1; i <= n; i++) {
		cin >> w[i];
		sum[get(i)] += w[i]; //该块的和加上w[i]
	}
	char op;
	int l, r, d;

	while (m--) {
		cin >> op >> l >> r;
		if (op == 'C') {
			cin >> d;
			change(l, r, d);
		} else {
			cout << query(l, r) << endl;
		}
	}
	return 0;
}