15. 三数之和 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]
]

Code

func threeSum(nums []int) [][]int {
	if nums == nil || len(nums) < 3 {
		return [][]int{}
	}
	sort.Ints(nums)  
	result := [][]int{}

	for k, v := range nums {
		if k > 0 && nums[k] == nums[k-1] {// 当遇到遍历的时候第一个数据数据重复了，那么就直接跳过
			continue
		}

		left, r := k+1, len(nums)-1 

		for left < r {
			sum := nums[left] + nums[r] + v  

			if sum == 0 {  
				sum := []int{nums[left], nums[r], v}
				result = append(result, sum)

				for left < r && nums[left] == nums[left+1]{ // 这里是观察后面的 left和r 往前和往后是否还是有重复的数据如果有的话直接就跳过即可
					left++
				}
				for left < r && nums[r] == nums[r-1]{  
					r--
				}
				left++ 
				r--
			}else if sum < 0 {  
				left++
			}else if sum > 0 { 
				r--
			}
		}
	}

	return result
}