当前位置: 代码迷 >> 综合 >> LeetCode(golang):15、 三数之和
  详细解决方案

LeetCode(golang):15、 三数之和

热度:75   发布时间:2024-02-02 06:20:03.0

题目:

给你一个包含 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;}