题目链接
思路
排序+双指针+去重
代码
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
}