✨494.目标和

Author: techoc

README.md

@@ -175,6 +175,7 @@
 - [386.字典序排数](leetcode/medium/386.go)
 - [393.UTF-8 编码验证](leetcode/medium/393.go)
 - [429.N 叉树的层序遍历](leetcode/medium/429.go)
+- [494.目标和](leetcode/medium/t494/solution.go)
 - [513.找树左下角的值](leetcode/medium/513.go)
 - [535.TinyURL 的加密与解密](leetcode/medium/535.go)
 - [537.复数乘法](leetcode/medium/537.go)

leetcode/medium/t494/solution.go

@@ -0,0 +1,29 @@
+package t494
+
+// 494. 目标和
+// https://leetcode.cn/problems/target-sum/
+func findTargetSumWays(nums []int, target int) (count int) {
+	// 定义回溯函数，用于递归地计算所有可能的和
+	var backtrack func(int, int)
+	// 回溯函数，递归核心部分，用于计算给定索引位置的元素加减后与目标值的匹配情况。
+	// 参数:
+	//   index - 当前处理的元素索引。
+	//   sum - 当前路径上的和。
+	backtrack = func(index, sum int) {
+		// 如果索引达到数组长度，说明遍历完所有元素
+		if index == len(nums) {
+			// 如果当前和等于目标值，增加计数
+			if sum == target {
+				count++
+			}
+			return
+		}
+		// 递归调用，分别尝试加法和减法
+		backtrack(index+1, sum+nums[index])
+		backtrack(index+1, sum-nums[index])
+	}
+	// 从数组的第一个元素开始回溯，初始和为0
+	backtrack(0, 0)
+	// 返回达到目标和的方法总数
+	return
+}

leetcode/medium/t494/solution_test.go

@@ -0,0 +1,21 @@
+package t494
+
+import "testing"
+
+func TestFindTargetSumWays(t *testing.T) {
+	tests := []struct {
+		nums   []int
+		target int
+		want   int
+	}{
+		{[]int{1, 1, 1, 1, 1}, 3, 5},
+		{[]int{1}, 1, 1},
+	}
+	for _, tt := range tests {
+		t.Run("", func(t *testing.T) {
+			if got := findTargetSumWays(tt.nums, tt.target); got != tt.want {
+				t.Errorf("findTargetSumWays() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}