Add LeetCode Weekly Contest 273

Author: EndlessCheng

leetcode/weekly/273/a/a.go

@@ -0,0 +1,8 @@
+package main
+
+// 一行
+
+// github.com/EndlessCheng/codeforces-go
+func isSameAfterReversals(num int) bool {
+	return num == 0 || num%10 > 0
+}

leetcode/weekly/273/a/a_test.go

@@ -0,0 +1,21 @@
+package main
+
+// github.com/EndlessCheng/codeforces-go
+var dirs = [][2]int{'L': {0, -1}, 'R': {0, 1}, 'U': {-1, 0}, 'D': {1, 0}}
+
+func executeInstructions(n int, startPos []int, s string) []int {
+	ans := make([]int, len(s))
+	for i := range s {
+		ans[i] = len(s) - i
+		x, y := startPos[0], startPos[1]
+		for j, ch := range s[i:] {
+			x += dirs[ch][0]
+			y += dirs[ch][1]
+			if x < 0 || x >= n || y < 0 || y >= n {
+				ans[i] = j
+				break
+			}
+		}
+	}
+	return ans
+}

leetcode/weekly/273/b/b.go

@@ -0,0 +1,24 @@
+package main
+
+// O(n) 做法：哈希表 + 枚举相同元素
+
+// github.com/EndlessCheng/codeforces-go
+func getDistances(arr []int) []int64 {
+	pos := map[int][]int{}
+	for i, v := range arr {
+		pos[v] = append(pos[v], i) // 记录相同元素的位置
+	}
+	ans := make([]int64, len(arr))
+	for _, p := range pos {
+		sum := 0
+		for _, i := range p {
+			sum += i - p[0]
+		}
+		ans[p[0]] = int64(sum) // 最左侧元素的间隔和
+		for i, n := 1, len(p); i < n; i++ { // 计算下一个相同元素的间隔和
+			sum -= (n - i*2) * (p[i] - p[i-1]) // 到右边的 n-i 个点的距离更近了，同时到左边 i 个点的距离更远了
+			ans[p[i]] = int64(sum)
+		}
+	}
+	return ans
+}

leetcode/weekly/273/b/b_test.go

@@ -0,0 +1,34 @@
+package main
+
+import "sort"
+
+/* 枚举 higher[0] + 双指针
+
+将 $\textit{nums}$ 排序后，$\textit{lower}[0]$ 必然是 $\textit{nums}[0]$。我们可以在 $\textit{nums}$ 中枚举 $\textit{higher}[0]$ 的值，从而得到 $k=\dfrac{\textit{higher}[0]-\textit{lower}[0]}{2}$。
+
+由于 $\textit{higher}[i]-\textit{lower}[i]=2k$ 是个定值，我们可以用双指针去遍历 $\textit{nums}$，计算其余的 $\textit{lower}$ 和 $\textit{higher}$ 的元素值。
+
+细节见代码注释。
+
+*/
+
+// github.com/EndlessCheng/codeforces-go
+func recoverArray(nums []int) []int {
+	sort.Ints(nums)
+	for i, n := 1, len(nums); ; i++ {
+		d := nums[i] - nums[0]
+		if d == 0 || d&1 > 0 { continue } // 必须保证 k 是正整数
+		k := d / 2
+		vis := make([]bool, n) // 用来标记出现在 higher 中的数
+		vis[i] = true
+		ans := []int{(nums[0] + nums[i]) / 2}
+		for lo, hi := 0, i+1; hi < n; hi++ { // 双指针：lo 指向 lower，hi 指向 higher
+			for lo++; lo < n && vis[lo]; lo++ {} // 跳过出现在 higher 中的数
+			for ; hi < n && nums[hi]-nums[lo] < 2*k; hi++ {}
+			if hi == n || nums[hi]-nums[lo] > 2*k { break }
+			vis[hi] = true
+			ans = append(ans, (nums[lo]+nums[hi])/2)
+		}
+		if len(ans) == n/2 { return ans }
+	}
+}