双指针：15.三数之和

考点：双指针

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，
使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

• 使用双指针搜索之前排序
• 动态循环双指针m,n

package main

import (
    "fmt"
    "sort"
    "strconv"
    "strings"
)

func threeSum(nums []int) [][]int {
    var (
        listLen int
        left int
        right int

        retList [][]int
    )

    // 先排序，优化及排除对全部大于0的元素组合的遍历
    sort.Ints(nums)
    fmt.Println(nums)

    listLen = len(nums)
    for listIndex, listNum := range nums {
        right = listLen - 1
        left = listIndex + 1
        for left < right {
            // 排除全部大于0的组合
            if listNum > 0 {
                break
            }

            // 固定节点去重
            if (listIndex > 0 && nums[listIndex] == nums[listIndex - 1]) {
                left ++
                continue
            }

            // 左、右节点去重
            if (nums[left] + nums[right] + listNum) > 0 || (right + 1 < listLen && nums[right] == nums[right+1]){
                right --
                continue
            } else if (nums[left] + nums[right] + listNum) < 0 || (left > listIndex + 1 && nums[left] == nums[left-1]) {
                left ++
                continue
            } else if (nums[left] + nums[right] + listNum) == 0 {
                tmpList := []int{listNum, nums[left], nums[right]}
                retList = append(retList, tmpList)

                fmt.Println(listNum, nums[left], nums[right])
                left ++
            }
        }
    }

    return retList
}

func main() {
    var (
        sum_list string
        num_list []int
    )

    fmt.Println("输入数组:")
    fmt.Scan(&sum_list)

    list_str := strings.Split(sum_list, ",")
    for _, str := range list_str {
        tmp, _ := strconv.Atoi(str)
        num_list = append(num_list, tmp)
    }

    fmt.Println(threeSum(num_list))
}