【leetcode15】三数之和--go语言实现

题目描述

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 。请你找出所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

示例 2：

输入：nums = []
输出：[]

示例 3：

输入：nums = [0]
输出：[]

思路分析：

如果采用哈希表解决的话，去重问题比较难解决，所以这里使用双指针，先对数据进行排序，用一轮循环确定i的值，然后从i+1到最后一个元素采用双指针进行处理。这里需要注意的是去重问题，所以在指针移动的过程中，只要当前状态与上一次状态的值相同的时候，就不做处理，继续移动

代码实现：

func threeSum(nums []int) [][]int {
	res:=[][]int{}//记录返回的结果
	lens:=len(nums)
	if lens<=2{//如果当前元素没有三个的时候
		return res
	}
	sort.Ints(nums)//排序
	if nums[0]>0||nums[lens-1]<0{//全部为正数或者全部为负数的话是没有答案的
		return res
	}
	for i:=0;i<lens-1;i++{
		if nums[i]>0{//如果第一个元素都大于0的话，后面肯定不可能相加满足等于0，直接退出
			break
		}
		if i>0&&nums[i]==nums[i-1]{//去重处理
			continue
		}
		j,k:=i+1,lens-1//双指针
		for j<k{
			if j!=i+1&&nums[j]==nums[j-1]{//去重处理
				j++
				continue
			}
			if k!=lens-1&&nums[k]==nums[k+1]{//去重处理
				k--
				continue
			}
			if nums[i]+nums[j]+nums[k]<0{//当前三个值小于0，指针j移动
				j++
			}else if nums[i]+nums[j]+nums[k]>0{//当前三个值小大于0，指针k移动
				k--
			}else{//如果找到满足条件之后，加入到res数组，两个指针同时移动
				temp:=[]int{nums[i],nums[j],nums[k]}
				res=append(res,temp)
				j++
				k--
			}
		}
	}
	return res//返回结果
}