跳到主要内容

15. 三数之和

mid

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

image-20220227160232830

排序 + 对撞指针​

l 和 r 双指针从头/尾向中间移动

func threeSum(nums []int) [][]int {
	if len(nums) < 3 {
		return [][]int{}
	}
	sort.Ints(nums)
	res := make([][]int, 0)

	for i := 0; i < len(nums)-2; i++ {
		a := nums[i]
		if a > 0 {
			break
		}
		if i > 0 && nums[i] == nums[i-1] {
			// 避免i重复
			continue
		}
		l, r := i+1, len(nums)-1
		for l < r {
			b, c := nums[l], nums[r]
			if a+b+c > 0 {
				r--
			} else if a+b+c < 0 {
				l++
			} else {
				res = append(res, []int{a, b, c})
				// 避免l或r重复
				for l < r && nums[l] == b {
					l++
				}
				for r > l && nums[r] == c {
					r--
				}
			}
		}
	}
	return res
}