【Leetcode】15. 三数之和

题目链接

【Leetcode】15. 三数之和

思路

排序+双指针+去重

代码

func threeSum(nums []int) [][]int {
	// 切片升序排序
	sort.Slice(nums, func(i, j int) bool {
		return nums[i] < nums[j]
	})
    // 结果切片
	ans := [][]int{}
	n := len(nums)
    // 枚举第一个位置
	for i := 0; i < n-2; i++ {
        // 如果枚举的第一个数字 已经 > 0，则不存在符合条件的结果
        if nums[i] > 0 {
            break
        }
        // 对第一个数字相等的情况进行去重
		if i > 0 && nums[i] == nums[i-1] {
			continue
		}
		// 双指针
		l, r := i+1, n-1
		for l < r {
            // 当前的总和
			sum := nums[i] + nums[l] + nums[r]
            // 如果总和=0,则符合条件
			if sum == 0 {
                // 添加进结果数组
				ans = append(ans, []int{nums[i], nums[l], nums[r]})
                // 对第二个数字进行去重
				for l < r && nums[l] == nums[l+1] {
					l++
				}
                // 对第三个数字进行去重
				for l < r && nums[r] == nums[r-1] {
					r--
				}
                // 第二个数字跟第三个数字的指针移动
				l++
				r--
			} else if sum < 0 {
				l++
			} else {
				r--
			}
		}
	}
	return ans
}