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
• 你可以按任意顺序返回答案 。

排序 + 双指针

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

// 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
}