排序+双指针,注意剪枝的几种情况。
- 最小四个数比target大
- 选择的前两个数遇到重复数字时
- 剩下两数最大比
target-preSum
小或者最小比 target-preSum
大(也可以不判断)
- 剩下两数遇到重复数字时
时间复杂度:O(n^3)
空间复杂度:O(logN),快排的递归深度
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
| func fourSum(nums []int, target int) [][]int { sort.Ints(nums) n := len(nums) ans := [][]int{}
for i := 0; i < n - 3; i++ { if nums[i] + nums[i+1] + nums[i+2] + nums[i+3] > target{ continue }
if i > 0 && nums[i] == nums[i-1]{ continue }
for j := i + 1; j < n - 2; j++{ preSum := nums[i] + nums[j]
if j > i + 1 && nums[j] == nums[j-1]{ continue }
l, r := j + 1, n - 1 if preSum + nums[r] + nums[r-1] < target{ continue } if preSum + nums[l] + nums[l+1] > target{ continue }
for l < r { if preSum + nums[l] + nums[r] == target { ans = append(ans, []int{nums[i], nums[j], nums[l], nums[r]}) for l < r { l++ if nums[l-1] != nums[l] { break } } for l < r { r-- if nums[r+1] != nums[j] { break } } } if preSum + nums[l] + nums[r] < target{ l++ } if preSum + nums[l] + nums[r] > target{ r-- } } } }
return ans }
|