LeetCode 15 三数之和

题目:

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。

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

解题:

三层循环,加起来为0的数组合并起来

代码:

func threeSum(nums []int) [][]int {
    var s [][]int
    l := len(nums)
    if l < 3 {
        return s
    }
    for i := 0; i < l-2; i++ {
        for j := 1; j < l-1; j++ {
            for k := 2; k < l; k++ {
                if nums[i]+nums[j]+nums[k] == 0 {
                    s = append(s, []int{nums[i], nums[j], nums[k]})
                }
            }
        }
    }
    return s
}

提交没有通过,最终结果没有去重

官方解答:排序+双指针

func threeSum(nums []int) [][]int {
    n := len(nums)
    sort.Ints(nums)
    ans := make([][]int, 0)
    for first := 0; first < n; first++ {
        if first > 0 && nums[first] == nums[first-1] {
            continue
        }
        third := n - 1
        target := -1 * nums[first]
        for second := first + 1; second < n; second++ {
            if second > first+1 && nums[second] == nums[second-1] {
                continue
            }
            for second < third && nums[second]+nums[third] > target {
                third--
            }
            if second == third {
                break
            }
            if nums[second]+nums[third] == target {
                ans = append(ans, []int{nums[first], nums[second], nums[third]})
            }
        }
    }
    return ans
}

我终究还是没能把双指针用到实例中去

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注