Add LeetCode Biweekly Contest 67

Author: EndlessCheng

leetcode/biweekly/67/a/a.go

@@ -0,0 +1,20 @@
+package main
+
+import "sort"
+
+// 两次排序
+
+// github.com/EndlessCheng/codeforces-go
+func maxSubsequence(nums []int, k int) []int {
+	id := make([]int, len(nums))
+	for i := range id {
+		id[i] = i
+	}
+	sort.Slice(id, func(i, j int) bool { return nums[id[i]] > nums[id[j]] }) // 按元素值从大到小排序
+	sort.Ints(id[:k]) // 对下标从小到大排序
+	ans := make([]int, k)
+	for i, j := range id[:k] {
+		ans[i] = nums[j]
+	}
+	return ans
+}

leetcode/biweekly/67/a/a_test.go

@@ -0,0 +1,31 @@
+package main
+
+// 前后缀经典技巧
+
+// github.com/EndlessCheng/codeforces-go
+func goodDaysToRobBank(security []int, time int) (ans []int) {
+	n := len(security)
+	if time*2 >= n { // i 最小为 time，而且需要满足 i+time<n，即 time*2<n，不满足则直接返回空数组
+		return
+	}
+
+	suf := make([]int, n) // suf[i] 表示从 security[i+1] 向后的最长连续非递减长度
+	for i := n - 2; i >= 0; i-- {
+		if security[i] <= security[i+1] {
+			suf[i] = suf[i+1] + 1
+		}
+	}
+
+	pre := 0 // pre 表示从 security[i-1] 向前的最长连续非递增长度
+	for i, v := range security[:n-time] {
+		if i > 0 && v <= security[i-1] {
+			pre++
+		} else {
+			pre = 0
+		}
+		if i >= time && pre >= time && suf[i] >= time {
+			ans = append(ans, i)
+		}
+	}
+	return
+}

leetcode/biweekly/67/b/b.go

@@ -0,0 +1,35 @@
+package main
+
+// 建图 + 暴力枚举所有起点
+
+// github.com/EndlessCheng/codeforces-go
+func maximumDetonation(bombs [][]int) (ans int) {
+	n := len(bombs)
+	g := make([][]int, n)
+	for i, p := range bombs {
+		for j, q := range bombs {
+			if j != i && (q[0]-p[0])*(q[0]-p[0])+(q[1]-p[1])*(q[1]-p[1]) <= p[2]*p[2] {
+				g[i] = append(g[i], j) // 有向图
+			}
+		}
+	}
+	for i := range g {
+		vis := make([]bool, n)
+		cnt := 0
+		var dfs func(int)
+		dfs = func(v int) {
+			vis[v] = true
+			cnt++
+			for _, w := range g[v] {
+				if !vis[w] {
+					dfs(w)
+				}
+			}
+		}
+		dfs(i)
+		if cnt > ans {
+			ans = cnt
+		}
+	}
+	return
+}

leetcode/biweekly/67/b/b_test.go

@@ -0,0 +1,55 @@
+package main
+
+import "github.com/emirpasic/gods/trees/redblacktree"
+
+/* 巧妙利用查询的特殊性
+
+为了高效插入数据，我们可以用平衡树来维护所有景点。
+
+注意到每次 `get` 后仅会将查询次数加一，我们可以直接用一个迭代器（指针）$\textit{cur}$ 指向下次查询需要返回的元素。
+
+- 对于添加操作，如果添加的景点排在当前 $\textit{cur}$ 前面，那么移动 $\textit{cur}$ 至其前一个元素，否则不移动 $\textit{cur}$；
+- 对于查询操作，每次查询结束后将 $\textit{cur}$ 移至其下一个元素。
+
+代码实现时，可以在初始时插入一个哨兵元素，从而简化判断逻辑。
+
+*/
+
+// github.com/EndlessCheng/codeforces-go
+type pair struct {
+	score int
+	name  string
+}
+
+func compare(x, y interface{}) int {
+	a, b := x.(pair), y.(pair)
+	if a.score > b.score || a.score == b.score && a.name < b.name {
+		return -1
+	}
+	return 1
+}
+
+type SORTracker struct {
+	*redblacktree.Tree
+	cur redblacktree.Iterator
+}
+
+func Constructor() SORTracker {
+	root := redblacktree.NewWith(compare)
+	root.Put(pair{}, nil) // 哨兵
+	return SORTracker{root, root.IteratorAt(root.Left())}
+}
+
+func (t *SORTracker) Add(name string, score int) {
+	p := pair{score, name}
+	t.Put(p, nil)
+	if compare(p, t.cur.Key()) < 0 {
+		t.cur.Prev() // 移动至前一个元素
+	}
+}
+
+func (t *SORTracker) Get() string {
+	name := t.cur.Key().(pair).name
+	t.cur.Next() // 移动至下一个元素
+	return name
+}