题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
分析:
本题首先想到的是用暴力解法,采用三层循环,将所有符合条件的情况都列出来,如下:
func threeSum(nums []int) [][]int { 长度 l := len(nums) re := [][]int{} for i := 0; i < l - 2; i++ { for j := i + 1; j < l - 1; j++ { for z := j + 1; z < l; z++ { if nums[i] + nums[j] + nums[z] == 0 { a := []int{nums[i],nums[j],nums[z]} re = append(re,a) } } } } return re }
但是这样并不能解决有重复结果的情况,是错误的。
解法一
分析:
本题采用排序+双指针的方法,先将数组从小到大排序,在用双指针从两边向中间遍历,
查找情况。先固定k位置的指针在最前面nums[k],i = k + 1, j = len(nums) ,开始遍历移动
i 和 j 指针, 如果nums[i] + nums[j] + nums[k] == 0 就是符合要求的数据组合,存到切片中。
nums[i] + nums[j] + nums[k] > 0 时 j -- ,否则一直 i ++ ,当 i > k + 1 ,nums[i] == nums[i - 1]
i 指针接着向后移动一位,防止重复组合。
答案:
func threeSum(nums []int) [][]int {
l := len(nums)
//进行排序
sort.Ints(nums)
result := make([][]int, 0)
//为了防止有重复的组合,有相同的数就跳过
for k := 0; k < l; k++ {
if k > 0 && nums[k] == nums[k - 1] {
continue;
}
if nums[k] > 0 {
break
}
//最右边的指针
j := l-1
//中间指针,枚举中间的数
for i := k + 1; i < l; i++ {
//需要和上次枚举的数不相同
if i > k + 1 && nums[i] == nums[i - 1] {
continue
}
//第二个指针在第三个指针的左侧,并且在三个数相加的时候大于0的时候
//就将第三个指针左移
for i < j && nums[i] + nums[j] + nums[k] >0 {
j--
}
//指针重合,退出循环
if i == j {
break;
}
if nums[i] + nums[j] + nums[k] == 0 {
result = append(result,[]int{nums[i],nums[j],nums[k]})
}
}
}
return result;
}