15. 3Sum

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

找出一个数组中，和为0的三个元素的集合，且不能重复。

func threeSum(nums []int) [][]int {
    var results [][]int
    if len(nums) < 3 {
        return results
    }
    sort.Ints(nums)
    for i := 0; i < len(nums) && nums[i] <= 0; i++ {
        if i > 0 && nums[i] == nums[i-1] {
            continue
        }
        j := i + 1
        k := len(nums) - 1
        for j < k {
            sum := nums[i] + nums[j] + nums[k]
            if sum == 0 {
                results = append(results, []int{nums[i], nums[j], nums[k]})
                for j < len(nums) - 1 && nums[j] == nums[j+1] {
                    j++
                }
                for k > 0 && nums[k] == nums[k-1] {
                    k--
                }
                j++
                k--
            } else if sum < 0 {
                j++
            } else if sum > 0 {
                k--
            }
        }
    }
    return results
}