题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
解法:
解法一:三重循环
该方法比较直观,但是会超时,对方法一优化得到方法二
import "sort"
func threeSum(nums []int) [][]int {
ans := make([][]int, 0)
sort.Ints(nums)
for i := 0; i < len(nums); i ++ {
if 0 < i && nums[i] == nums[i - 1] {
continue
}
for j := i + 1; j < len(nums); j ++ {
// 下面这个条件去重
if j != i + 1 && nums[j] == nums[j - 1] {
continue
}
for k := j + 1; k < len(nums); k ++ {
// 下面这个条件去重
if k != j + 1 && nums[k] == nums[k - 1] {
continue
}
if nums[i] + nums[j] + nums[k] == 0 {
ans = append(ans, []int{nums[i], nums[j], nums[k]})
}
}
}
}
return ans
}
方法二:双指针
import "sort"
func threeSum(nums []int) [][]int {
if len(nums) < 3 {
return nil
}
ans := make([][]int, 0)
sort.Ints(nums)
for i := 0; i < len(nums); i ++ {
if 0 < i && nums[i] == nums[i - 1] {
continue
}
k := len(nums) - 1
for j := i + 1; j < len(nums); j ++ {
if i + 1 < j && nums[j] == nums[j - 1] {
continue
}
for j < k {
sum := nums[i] + nums[j] + nums[k]
if sum > 0 {
k --
} else if sum < 0 {
break
} else {
if k + 1 < len(nums) && nums[k] == nums[k + 1] {
break
}
ans = append(ans, []int{nums[i], nums[j], nums[k]})
break
}
}
}
}
return ans
}