每日一题---15. 三数之和[力扣][Go]

题目描述

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

解题代码

失败

func threeSum(nums []int) [][]int {
// 想着利用双指针从两头开始逐个遍历，
// 让一个指针在两个指针间来回移动，
// 但是这样就会有遍历不到的数组
	//flag := false
	ints := [][]int{}
	intss := []int{}
	sort.Ints(nums)
	var moveIndex int
	leftIndex := 0
	rightIndex := len(nums) - 1
	index := 0
	for leftIndex < rightIndex {
		if nums[leftIndex]*nums[rightIndex] > 0 {
			return ints
		}
		moveIndex = leftIndex + 1
		for moveIndex < rightIndex {
			if nums[leftIndex] + nums[rightIndex] + nums[moveIndex] == 0 {
				//if nums[leftIndex] == 0 && nums[rightIndex] == 0{
				//	if flag {
				//		break
				//	}
				//	flag = true
				//}
				ints = append(ints, append(intss, nums[leftIndex],nums[rightIndex],nums[moveIndex]))
				index++
				break
			}
			moveIndex ++
		}
		moveIndexs(&leftIndex,&rightIndex,nums)
	}
	return ints
}

func moveIndexs(l,r *int,num []int) {
	if int(math.Abs(float64(num[*l]))) > num[*r] {
		*l ++
		for num[*l] == num[*l-1] {
			*l++
			if *l >= *r {
				return
			}
		}
	} else {
		*r --
		for num[*r] == num[*r+1] {
			if *l >= *r {
				return
			}
			*r --
		}
	}
}

成功

// 评论区 @代码随想录 大佬的解题过程
func threeSum(nums []int)[][]int{
	sort.Ints(nums)
	res:=[][]int{}

	for i:=0;i<len(nums)-2;i++{
// 先固定一头，然后用双指针法一个一个过
		n1:=nums[i]
		if n1>0{
			break
		}
		if i>0&&n1==nums[i-1]{
			continue
		}
		l,r:=i+1,len(nums)-1
		for l<r{
			n2,n3:=nums[l],nums[r]
			if n1+n2+n3==0{
				res=append(res,[]int{n1,n2,n3})
				for l<r&&nums[l]==n2{
					l++
				}
				for l<r&&nums[r]==n3{
					r--
				}
			}else if n1+n2+n3<0{
				l++
			}else {
				r--
			}
		}
	}
	return res
}

提交结果