-
Notifications
You must be signed in to change notification settings - Fork 674
/
Copy pathp2068.go
76 lines (67 loc) · 1.27 KB
/
p2068.go
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
74
75
76
package main
import (
"bufio"
. "fmt"
"io"
"math/bits"
)
// https://space.bilibili.com/206214
type seg2068 []struct{ l, r, val int }
func (t seg2068) build(o, l, r int) {
t[o].l, t[o].r = l, r
if l == r {
return
}
m := (l + r) >> 1
t.build(o<<1, l, m)
t.build(o<<1|1, m+1, r)
t.maintain(o)
}
func (t seg2068) update(o, i int, val int) {
if t[o].l == t[o].r {
t[o].val += val
return
}
m := (t[o].l + t[o].r) >> 1
if i <= m {
t.update(o<<1, i, val)
} else {
t.update(o<<1|1, i, val)
}
t.maintain(o)
}
func (t seg2068) maintain(o int) {
t[o].val = t[o<<1].val + t[o<<1|1].val
}
func (t seg2068) query(o, l, r int) (res int) {
if l <= t[o].l && t[o].r <= r {
return t[o].val
}
m := (t[o].l + t[o].r) >> 1
if r <= m {
return t.query(o<<1, l, r)
}
if m < l {
return t.query(o<<1|1, l, r)
}
return t.query(o<<1, l, r) + t.query(o<<1|1, l, r)
}
func p2068(_r io.Reader, _w io.Writer) {
in := bufio.NewReader(_r)
out := bufio.NewWriter(_w)
defer out.Flush()
var n, m, l, r int
var op string
Fscan(in, &n, &m)
t := make(seg2068, 2<<bits.Len(uint(n-1)))
t.build(1, 1, n)
for ; m > 0; m-- {
Fscan(in, &op, &l, &r)
if op == "x" {
t.update(1, l, r)
} else {
Fprintln(out, t.query(1, l, r))
}
}
}
//func main() { p2068(os.Stdin, os.Stdout) }