18. 四数之和

mid

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按任意顺序返回答案。

排序 + 双指针

在三数之和的基础上，再套一层循环

 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
// fourSum 排序+双指针 外加一层循环
func fourSum(nums []int, target int) [][]int {
   sort.Ints(nums)
   curNums := make([][]int, 0)

   for j := 0; j < len(nums); j++ {
      if j > 0 && nums[j] == nums[j-1] {
         continue
      }
      for i := j + 2; i < len(nums); i++ {
         start, end := j+1, len(nums)-1
         // 考虑i跟前一个i相同的情况
         if i > j+2 && nums[i] == nums[i-1] {
            start = i - 1
         }

         for start < i && end > i {
            // 去重
            if start > j+1 && nums[start] == nums[start-1] {
               start++
               continue
            }
            if end < len(nums)-1 && nums[end] == nums[end+1] {
               end--
               continue
            }

            curSum := nums[j] + nums[start] + nums[i] + nums[end]
            switch {
            case curSum == target:
               curNums = append(curNums, []int{nums[j], nums[start], nums[i], nums[end]})
               start++
               end--
            case curSum < target:
               start++
            default:
               end--
            }
         }
      }
   }
   return curNums
}