LeetCode(golang):15、 三数之和

题目:

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

}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值