给出n个数字,有两个操作,一个是查询某个位置上的数值,另一个是给定区间[l,r]和一个数字d,要求从l到r中的每个位置i开始,a[i]增加d,a[2*i],a[3*i], … ,每个位置都增加d。
方法是采用分桶法和平方分割,也被称为块状数组,分成a个有floor(sqrt(n))个元素的桶,每个桶维护各自的信息,即增加了多少。
查询时,枚举因子求出这个数被增加了多少,再与原数相加即可。
#include <cstdio>
#include <vector>
using namespace std;
const int maxn = 300000 + 5;
const int b = 600;
int a[maxn];
long long num[maxn/b];
// vector<long long> bucket[maxn/b];
long long bucket[maxn/b][b];
int n, m;
int main(int argc, char const *argv[]) {
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%d", &a[i]);
// bucket[i/b].push_back(0LL);
}
scanf("%d", &m);
while (m--) {
int op;
scanf("%d", &op);
if (op == 1) {
int x;
scanf("%d", &x);
long long ans = a[x-1];
for (int i = 1; i * i <= x; i++) {
if (x % i == 0) {
ans += bucket[(i-1)/b][(i-1)%b] + num[(i-1)/b];
if (x / i != i) {
int j = x / i;
ans += bucket[(j-1)/b][(j-1)%b] + num[(j-1)/b];
}
}
}
printf("%lld\n", ans);
} else {
int l, r, d;
scanf("%d%d%d", &l, &r, &d);
l--, r--;
int s = l / b, t = r / b;
if (s < t) {
// for (int i = l % b; i < bucket[s].size(); i++) {
for (int i = l % b; i < b; i++) {
bucket[s][i] += d;
}
for (int i = s + 1; i < t; i++) {
num[i] += d;
}
for (int i = 0; i <= r % b; i++) {
bucket[t][i] += d;
}
} else {
for (int i = l % b; i <= r % b; i++) {
bucket[s][i] += d;
}
}
}
}
return 0;
}