15三数之和解题记录
看到题目标签排序+双指针时,由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 时处理过,这样会导致重复解。