看到题目标签排序+双指针时,由11接雨水遍历宽的思路想到了办法。先排序,然后双重遍历,外重遍历整个数组,内重遍历双指针寻找能组合为0的另外两个数。但是由于避免输出的重复,外重遍历时需要跳过相同的数字。这道题的难点在于对结果的去重。
func threeSum(nums []int) [][]int {sort.Ints(nums) // 先排序res := make([][]int, 0)n := len(nums)for i := 0; i < n; i++ { // 外层遍历,固定第一个数if i > 0 && nums[i] == nums[i-1] { // 跳过重复元素continue}target := -nums[i] // 转换为两数之和问题left, right := i+1, n-1 // 双指针for left < right {sum := nums[left] + nums[right]if sum == target { // 找到解res = append(res, []int{nums[i], nums[left], nums[right]})// 跳过重复的 left 和 right (这个最开始的思路遗漏了)for left < right && nums[left] == nums[left+1] {left++}for left < right && nums[right] == nums[right-1] { // 判断left < right 是可能发生交叉后访问越界right--}left++right--} else if sum < target { // 太小,左指针右移left++} else { // 太大,右指针左移right--}}}return res
}
代码中还有一个关键的:left 从 i+1 开始而不是 0,避免重复解。假设 nums = [-1, 0, 1, 2, -1, -4],排序后是 [-4, -1, -1, 0, 1, 2]。当 i=1(nums[i] = -1),如果 left 可以从 0 开始,会重新计算 nums[0] = -4 的组合,但 nums[0] 已经在 i=0 时处理过,这样会导致重复解。