15. 三数之和

howeres

已于 2023-05-30 00:06:37 修改

阅读量81

点赞数

文章标签：算法数据结构 leetcode

于 2023-05-29 23:41:23 首次发布

本文链接：https://blog.csdn.net/howeres/article/details/130938772

版权

该代码实现了一个函数，用于解决给定数组中找到所有不重复的三数之和为零的组合。首先对数组进行排序，然后使用三指针法（夹心饼干策略）遍历数组，通过调整起始和结束指针来找到合适的配对。在过程中，为了避免重复组合，对重复数字进行了特殊处理。

摘要由CSDN通过智能技术生成

Given an integer array nums, return all the triplets

Thoughts:

由于需要去重, 需要先排序
声明所有的变量, 避免写代码时输入冒号colon
由于是三数之和, 我们采用夹心饼干的方式, 从第2位 to 倒数第2位, 作为基数
在确定好基数之后, 我们再采用双指针的方式, 确定第1和第3位的数字
处理剩下细节, 重复的补数可以跳过, 如果基数重复则前指针全已知, 后指针还需重新计算 (这么做的目的是因为数字可以重, 但位置不能重), 符合的数组加入ans, 前后各缩一次
nums已经排序过, 所以当nums[start] > 0时, 后面无需判断, 可以直接进入下一个基数

func threeSum(nums []int) [][]int {
	sort.Ints(nums)
	ans, start, end, index, addNum, length := make([][]int, 0), 0, 0, 0, 0, len(nums)
	outloop:
	for  index = 1; index < length-1; index++ {
		start, end = 0, length-1
		if index > 1 && nums[index] == nums[index-1] {
			start = index - 1
		}
		for start < index && end > index {
			if nums[start] > 0 {
				continue outloop
			}
			if start > 0 && nums[start] == nums[start-1] {
				start++
				continue
			}
			if end < length-1 && nums[end] == nums[end+1] {
				end--
				continue
			}
			addNum = nums[start] + nums[end] + nums[index]
			if addNum == 0 {
				ans = append(ans, []int{nums[start], nums[index], nums[end]})
				start++
				end--
			} else if addNum > 0 {
				end--
			} else {
				start++
			}
		}
	}
	return ans
}