【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//返回结果
}

猜你喜欢

转载自blog.csdn.net/weixin_42918559/article/details/125112779