15. 三数之和

Given an integer array nums, return all the triplets

Thoughts:

  1. 由于需要去重, 需要先排序
  2. 声明所有的变量, 避免写代码时输入冒号colon
  3. 由于是三数之和, 我们采用夹心饼干的方式, 从第2位 to 倒数第2位, 作为基数
  4. 在确定好基数之后, 我们再采用双指针的方式, 确定第1和第3位的数字
  5. 处理剩下细节, 重复的补数可以跳过, 如果基数重复则前指针全已知, 后指针还需重新计算 (这么做的目的是因为数字可以重, 但位置不能重), 符合的数组加入ans, 前后各缩一次
  6. nums已经排序过, 所以当nums[start] > 0时, 后面无需判断, 可以直接进入下一个基数
func threeSum(nums []int) [][]int {
	sort.Ints(nums)
	ans, start, end, index, addNum, length := make([][]int, 0), 0, 0, 0, 0, len(nums)
	outloop:
	for  index = 1; index < length-1; index++ {
		start, end = 0, length-1
		if index > 1 && nums[index] == nums[index-1] {
			start = index - 1
		}
		for start < index && end > index {
			if nums[start] > 0 {
				continue outloop
			}
			if start > 0 && nums[start] == nums[start-1] {
				start++
				continue
			}
			if end < length-1 && nums[end] == nums[end+1] {
				end--
				continue
			}
			addNum = nums[start] + nums[end] + nums[index]
			if addNum == 0 {
				ans = append(ans, []int{nums[start], nums[index], nums[end]})
				start++
				end--
			} else if addNum > 0 {
				end--
			} else {
				start++
			}
		}
	}
	return ans
}

你可能感兴趣的:(算法,数据结构,leetcode)