|
| 1 | ++++ |
| 2 | +title = '18. 四数之和' |
| 3 | +date = 2024-05-01T10:14:47+08:00 |
| 4 | +draft = false |
| 5 | + |
| 6 | ++++ |
| 7 | + |
| 8 | +## [题目](https://leetcode.cn/problems/4sum/) |
| 9 | + |
| 10 | +给你一个由 `n` 个整数组成的数组 `nums` ,和一个目标值 `target` 。请你找出并返回满足下述全部条件且**不重复**的四元组 `[nums[a], nums[b], nums[c], nums[d]]` (若两个四元组元素一一对应,则认为两个四元组重复): |
| 11 | + |
| 12 | +- `0 <= a, b, c, d < n` |
| 13 | +- `a`、`b`、`c` 和 `d` **互不相同** |
| 14 | +- `nums[a] + nums[b] + nums[c] + nums[d] == target` |
| 15 | + |
| 16 | +你可以按 **任意顺序** 返回答案 。 |
| 17 | + |
| 18 | +**示例 1:** |
| 19 | + |
| 20 | +``` |
| 21 | +输入:nums = [1,0,-1,0,-2,2], target = 0 |
| 22 | +输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]] |
| 23 | +``` |
| 24 | + |
| 25 | +**示例 2:** |
| 26 | + |
| 27 | +``` |
| 28 | +输入:nums = [2,2,2,2,2], target = 8 |
| 29 | +输出:[[2,2,2,2]] |
| 30 | +``` |
| 31 | + |
| 32 | +**提示:** |
| 33 | + |
| 34 | +- `1 <= nums.length <= 200` |
| 35 | +- `-109 <= nums[i] <= 109` |
| 36 | +- `-109 <= target <= 109` |
| 37 | + |
| 38 | +## 解题思路 |
| 39 | + |
| 40 | +方法一:排序+双指针,在[3Sum](/leetcode-problems/3sum)的基础上多套一层循环。第一层循环固定第一个元素,第二层循环固定第二个元素,循环内部用对撞指针找出剩余两个元素。时间复杂度:$O(n^3)$,空间复杂度:$O(logn)$,取决于排序所需的空间。 |
| 41 | + |
| 42 | +方法二:分治法,将问题分成若干子问题,对子问题求解后将解合并。4Sum -> n-3个3Sum,3Sum -> n-2个2Sum,2Sum用对撞指针得到解。延展到求kSum。 |
| 43 | + |
| 44 | +## 代码 |
| 45 | + |
| 46 | +方法一 |
| 47 | + |
| 48 | +```go |
| 49 | +func fourSum(nums []int, target int) [][]int { |
| 50 | + var ans [][]int |
| 51 | + n := len(nums) |
| 52 | + sort.Ints(nums) |
| 53 | + for i := 0; i < n-3; i++ { |
| 54 | + if i > 0 && nums[i] == nums[i-1] { |
| 55 | + continue |
| 56 | + } |
| 57 | + for j := i+1; j < n-2; j++ { |
| 58 | + if j > i+1 && nums[j] == nums[j-1] { |
| 59 | + continue |
| 60 | + } |
| 61 | + l, r := j+1, n-1 |
| 62 | + for l < r { |
| 63 | + s := nums[i] + nums[j] + nums[l] + nums[r] |
| 64 | + if s == target { |
| 65 | + ans = append(ans, []int{nums[i], nums[j], nums[l], nums[r]}) |
| 66 | + lv, rv := nums[l], nums[r] |
| 67 | + for l < r && nums[l] == lv { |
| 68 | + l++ |
| 69 | + } |
| 70 | + for l < r && nums[r] == rv { |
| 71 | + r-- |
| 72 | + } |
| 73 | + } else if s < target { |
| 74 | + l++ |
| 75 | + } else { |
| 76 | + r-- |
| 77 | + } |
| 78 | + } |
| 79 | + } |
| 80 | + } |
| 81 | + return ans |
| 82 | +} |
| 83 | +``` |
| 84 | + |
| 85 | +方法二 |
| 86 | + |
| 87 | +```go |
| 88 | +func fourSum(nums []int, target int) [][]int { |
| 89 | + sort.Ints(nums) |
| 90 | + return kSum(nums, target, 4) |
| 91 | +} |
| 92 | + |
| 93 | +func kSum(nums []int, target int, k int) [][]int { |
| 94 | + // 剪枝 |
| 95 | + if !(nums[0]*k <= target && target <= nums[len(nums)-1]*k) { |
| 96 | + return [][]int{} |
| 97 | + } |
| 98 | + |
| 99 | + if k == 2 { |
| 100 | + return twoSum(nums, target) |
| 101 | + } |
| 102 | + |
| 103 | + var ans [][]int |
| 104 | + for i := 0; i < len(nums)-k+1; i++ { |
| 105 | + if i > 0 && nums[i] == nums[i-1] { |
| 106 | + continue |
| 107 | + } |
| 108 | + for _, res := range kSum(nums[i+1:], target-nums[i], k-1) { |
| 109 | + ans = append(ans, append([]int{nums[i]}, res...)) |
| 110 | + } |
| 111 | + } |
| 112 | + return ans |
| 113 | +} |
| 114 | + |
| 115 | +func twoSum(nums []int, target int) [][]int { |
| 116 | + var ans [][]int |
| 117 | + l, r := 0, len(nums)-1 |
| 118 | + for l < r { |
| 119 | + s := nums[l] + nums[r] |
| 120 | + if s == target { |
| 121 | + ans = append(ans, []int{nums[l], nums[r]}) |
| 122 | + lv, rv := nums[l], nums[r] |
| 123 | + for l < r && nums[l] == lv { |
| 124 | + l++ |
| 125 | + } |
| 126 | + for l < r && nums[r] == rv { |
| 127 | + r-- |
| 128 | + } |
| 129 | + } else if s < target { |
| 130 | + l++ |
| 131 | + } else { |
| 132 | + r-- |
| 133 | + } |
| 134 | + } |
| 135 | + return ans |
| 136 | +} |
| 137 | +``` |
| 138 | + |
0 commit comments