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
}