三数之和的golang实现

来源：cnblogs　　作者：timliudream　　时间：2018/12/17 9:45:05　　对本文有异议

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

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

我们来看一下图：

我们先对数组排序，得到如下图的结果

我们要计算a+b+c=0，先对数组循环得到a，然后b就是a的索引+1的值，c是len(nums)-1的值。

如果a+b+c<0的是时候，说明太小了，那b就往右挪一下

反之，如果a+b+c>0，那么说明太大了，c就往左挪一下

核心代码：

func threeSum2(nums []int) [][]int {
    //先对数组排序
    sort.Ints(nums)
    result := [][]int{}
    for i := 0; i < len(nums)-1; i++ {
        if i > 0 && nums[i] == nums[i-1] {
            continue
        }
        j := i + 1
        z := len(nums) - 1
        for z > j {
            b := nums[j]
            c := nums[z]
            if nums[i]+b+c > 0 {
                z--
            } else if nums[i]+b+c < 0 {
                j++
            } else {
                item := []int{nums[i], b, c}
                result = append(result, item)
                for j < z && nums[j] == nums[j+1] {
                    j++
                }
                for j < z && nums[z] == nums[z-1] {
                    z--
                }
                j++
                z--
            }
        }
    }
    return result
}