feat: add golang solution to lc problem: No.0699

No.0699.Falling Squares

Author: maolonglong

solution/0600-0699/0699.Falling Squares/README.md

@@ -393,6 +393,113 @@ public:
 };
 ```
 
+### **Go**
+
+```go
+type node struct {
+	left      *node
+	right     *node
+	l, mid, r int
+	v, add    int
+}
+
+func newNode(l, r int) *node {
+	return &node{
+		l:   l,
+		r:   r,
+		mid: int(uint(l+r) >> 1),
+	}
+}
+
+func max(x, y int) int {
+	if x > y {
+		return x
+	}
+	return y
+}
+
+type segmentTree struct {
+	root *node
+}
+
+func newSegmentTree() *segmentTree {
+	return &segmentTree{
+		root: newNode(1, 1e9),
+	}
+}
+
+func (t *segmentTree) modify(l, r, v int, n *node) {
+	if l > r {
+		return
+	}
+	if n.l >= l && n.r <= r {
+		n.v = v
+		n.add = v
+		return
+	}
+	t.pushdown(n)
+	if l <= n.mid {
+		t.modify(l, r, v, n.left)
+	}
+	if r > n.mid {
+		t.modify(l, r, v, n.right)
+	}
+	t.pushup(n)
+}
+
+func (t *segmentTree) query(l, r int, n *node) int {
+	if l > r {
+		return 0
+	}
+	if n.l >= l && n.r <= r {
+		return n.v
+	}
+	t.pushdown(n)
+	v := 0
+	if l <= n.mid {
+		v = max(v, t.query(l, r, n.left))
+	}
+	if r > n.mid {
+		v = max(v, t.query(l, r, n.right))
+	}
+	return v
+}
+
+func (t *segmentTree) pushup(n *node) {
+	n.v = max(n.left.v, n.right.v)
+}
+
+func (t *segmentTree) pushdown(n *node) {
+	if n.left == nil {
+		n.left = newNode(n.l, n.mid)
+	}
+	if n.right == nil {
+		n.right = newNode(n.mid+1, n.r)
+	}
+	if n.add != 0 {
+		n.left.add = n.add
+		n.right.add = n.add
+		n.left.v = n.add
+		n.right.v = n.add
+		n.add = 0
+	}
+}
+
+func fallingSquares(positions [][]int) []int {
+	ans := make([]int, len(positions))
+	t := newSegmentTree()
+	mx := 0
+	for i, p := range positions {
+		l, w, r := p[0], p[1], p[0]+p[1]-1
+		h := t.query(l, r, t.root) + w
+		mx = max(mx, h)
+		ans[i] = mx
+		t.modify(l, r, h, t.root)
+	}
+	return ans
+}
+```
+
 ### **...**
 
 ```

solution/0600-0699/0699.Falling Squares/README_EN.md

@@ -344,6 +344,113 @@ public:
 };
 ```
 
+### **Go**
+
+```go
+type node struct {
+	left      *node
+	right     *node
+	l, mid, r int
+	v, add    int
+}
+
+func newNode(l, r int) *node {
+	return &node{
+		l:   l,
+		r:   r,
+		mid: int(uint(l+r) >> 1),
+	}
+}
+
+func max(x, y int) int {
+	if x > y {
+		return x
+	}
+	return y
+}
+
+type segmentTree struct {
+	root *node
+}
+
+func newSegmentTree() *segmentTree {
+	return &segmentTree{
+		root: newNode(1, 1e9),
+	}
+}
+
+func (t *segmentTree) modify(l, r, v int, n *node) {
+	if l > r {
+		return
+	}
+	if n.l >= l && n.r <= r {
+		n.v = v
+		n.add = v
+		return
+	}
+	t.pushdown(n)
+	if l <= n.mid {
+		t.modify(l, r, v, n.left)
+	}
+	if r > n.mid {
+		t.modify(l, r, v, n.right)
+	}
+	t.pushup(n)
+}
+
+func (t *segmentTree) query(l, r int, n *node) int {
+	if l > r {
+		return 0
+	}
+	if n.l >= l && n.r <= r {
+		return n.v
+	}
+	t.pushdown(n)
+	v := 0
+	if l <= n.mid {
+		v = max(v, t.query(l, r, n.left))
+	}
+	if r > n.mid {
+		v = max(v, t.query(l, r, n.right))
+	}
+	return v
+}
+
+func (t *segmentTree) pushup(n *node) {
+	n.v = max(n.left.v, n.right.v)
+}
+
+func (t *segmentTree) pushdown(n *node) {
+	if n.left == nil {
+		n.left = newNode(n.l, n.mid)
+	}
+	if n.right == nil {
+		n.right = newNode(n.mid+1, n.r)
+	}
+	if n.add != 0 {
+		n.left.add = n.add
+		n.right.add = n.add
+		n.left.v = n.add
+		n.right.v = n.add
+		n.add = 0
+	}
+}
+
+func fallingSquares(positions [][]int) []int {
+	ans := make([]int, len(positions))
+	t := newSegmentTree()
+	mx := 0
+	for i, p := range positions {
+		l, w, r := p[0], p[1], p[0]+p[1]-1
+		h := t.query(l, r, t.root) + w
+		mx = max(mx, h)
+		ans[i] = mx
+		t.modify(l, r, h, t.root)
+	}
+	return ans
+}
+```
+
 ### **...**
 
 ```

solution/0600-0699/0699.Falling Squares/Solution.go

@@ -0,0 +1,102 @@
+type node struct {
+	left      *node
+	right     *node
+	l, mid, r int
+	v, add    int
+}
+
+func newNode(l, r int) *node {
+	return &node{
+		l:   l,
+		r:   r,
+		mid: int(uint(l+r) >> 1),
+	}
+}
+
+func max(x, y int) int {
+	if x > y {
+		return x
+	}
+	return y
+}
+
+type segmentTree struct {
+	root *node
+}
+
+func newSegmentTree() *segmentTree {
+	return &segmentTree{
+		root: newNode(1, 1e9),
+	}
+}
+
+func (t *segmentTree) modify(l, r, v int, n *node) {
+	if l > r {
+		return
+	}
+	if n.l >= l && n.r <= r {
+		n.v = v
+		n.add = v
+		return
+	}
+	t.pushdown(n)
+	if l <= n.mid {
+		t.modify(l, r, v, n.left)
+	}
+	if r > n.mid {
+		t.modify(l, r, v, n.right)
+	}
+	t.pushup(n)
+}
+
+func (t *segmentTree) query(l, r int, n *node) int {
+	if l > r {
+		return 0
+	}
+	if n.l >= l && n.r <= r {
+		return n.v
+	}
+	t.pushdown(n)
+	v := 0
+	if l <= n.mid {
+		v = max(v, t.query(l, r, n.left))
+	}
+	if r > n.mid {
+		v = max(v, t.query(l, r, n.right))
+	}
+	return v
+}
+
+func (t *segmentTree) pushup(n *node) {
+	n.v = max(n.left.v, n.right.v)
+}
+
+func (t *segmentTree) pushdown(n *node) {
+	if n.left == nil {
+		n.left = newNode(n.l, n.mid)
+	}
+	if n.right == nil {
+		n.right = newNode(n.mid+1, n.r)
+	}
+	if n.add != 0 {
+		n.left.add = n.add
+		n.right.add = n.add
+		n.left.v = n.add
+		n.right.v = n.add
+		n.add = 0
+	}
+}
+
+func fallingSquares(positions [][]int) []int {
+	ans := make([]int, len(positions))
+	t := newSegmentTree()
+	mx := 0
+	for i, p := range positions {
+		l, w, r := p[0], p[1], p[0]+p[1]-1
+		h := t.query(l, r, t.root) + w
+		mx = max(mx, h)
+		ans[i] = mx
+		t.modify(l, r, h, t.root)
+	}
+	return ans
+}