Codeforces 438D 题解

线段树区间取模,区间查询。

维护区间最大值,更新时进行剪枝,如果 $max[l…r] < mod$,那么剪枝不进行更新。

复杂度分析

一个数一旦被修改,那么它的大小至少除以了2。

总的修改次数不超过 nlogn 次。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define ms(a,b) memset(a,b,sizeof(a))
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
using namespace std;
typedef long long ll;
const int maxn = 100000 + 5;

int n, m, a[maxn], mx[maxn << 2];;
ll sum[maxn << 2];

void pushup(int rt){
sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
mx[rt] = max(mx[rt << 1], mx[rt << 1 | 1]);
}
void build(int l, int r, int rt){
if (l == r) {
sum[rt] = mx[rt] = a[l]; return;
}
int m = l + r >> 1;
build(lson); build(rson);
pushup(rt);
}
void update(int i, int x, int l, int r, int rt){
if (l == r){
sum[rt] = mx[rt] = x; return;
}
int m = l + r >> 1;
if (i <= m) update(i, x, lson);
else update(i, x, rson);
pushup(rt);
}
void update(int L, int R, int mod, int l, int r, int rt){
if (L > r || l > R || mx[rt] < mod) return;
if (L <= l && r <= R && l == r){
sum[rt] = mx[rt] = mx[rt] % mod;
return;
}
int m = l + r >> 1;
update(L, R, mod, lson); update(L, R, mod, rson);
pushup(rt);
}
ll query(int L, int R, int l, int r, int rt){
if (L <= l && r <= R) return sum[rt];
int m = l + r >> 1; ll ans = 0;
if (L <= m) ans += query(L, R, lson);
if (R > m) ans += query(L, R, rson);
return ans;
}

int main(){
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) scanf("%d", a + i);
build(1, n, 1); int op, l, r, k, x;
while (m--){
scanf("%d", &op);
if (op == 1){
scanf("%d%d", &l, &r); printf("%I64d\n", query(l, r, 1, n, 1));
}
if (op == 2){
scanf("%d%d%d", &l, &r, &x);
update(l, r, x, 1, n, 1);
}
if (op == 3){
scanf("%d%d", &k, &x);
update(k, x, 1, n, 1);
}
}
return 0;
}