LeetCode每日一题2020.6.12 15.三数之和

Posted 2020-12-10 lalal

15. 三数之和

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

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

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

分析：

? 一、如果使用暴力法，时间复杂度会达到$O(n^3)$。我们可以使用双指针来优化循环。

? 二、接着要解决判断重复的三元组。我们对每一重循环加上一个条件，判断当前元素与前一个元素是否相同，这样就能起到判重的效果。但是注意下图的第三项，即右指针，由于当正好匹配一个三元组后，左指针会直接移动。因此对于右指针来说，并没有判重的必要。

代码（Golang）：

func threeSum(nums []int) [][]int {
	var ans [][]int
	n := len(nums)
	sort.Ints(nums)
	for i := 0; i < n - 2; i++ {
		if i > 0 && nums[i] == nums[i - 1] {
			continue
		}
		right := n - 1
		target := -nums[i]
		for left := i + 1; left < n - 1; left++ {
			if left > i + 1 && nums[left] == nums[left - 1] {
				continue
			}
			for left < right && nums[left] + nums[right] > target {
				right--
			}
			if left == right {
				break
			}
			if nums[left] + nums[right] == target {
				ans = append(ans, []int{nums[i], nums[left], nums[right]})
			}
		}
	}
	return ans
}

小结：

? 当题目必须要进行多重循环的时候，可以考虑使用双指针来进行优化。尤其当我们需要枚举数组中的两个元素时，如果我们发现随着第一个元素的递增，第二个元素是递增的，那么就可以使用双指针。