15. 三数之和

70 阅读1分钟

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

解法:
解法一:三重循环
该方法比较直观,但是会超时,对方法一优化得到方法二

import "sort"
func threeSum(nums []int) [][]int {
    ans := make([][]int, 0)
    sort.Ints(nums)
    for i := 0; i < len(nums); i ++ {
        if 0 < i && nums[i] == nums[i - 1] {
            continue
        }
        for j := i + 1; j < len(nums); j ++ {
            // 下面这个条件去重
            if j != i + 1 && nums[j] == nums[j - 1] {  
                continue
            }
            for k := j + 1; k < len(nums); k ++ {
               // 下面这个条件去重
                if k != j + 1 && nums[k] == nums[k - 1] {
                    continue
                }
                if nums[i] + nums[j] + nums[k] == 0 {
                    ans = append(ans, []int{nums[i], nums[j], nums[k]})
                }
            }
        }
    }
    return ans
}

方法二:双指针

import "sort"
func threeSum(nums []int) [][]int {
    if len(nums) < 3 {
        return nil
    }
    ans := make([][]int, 0)
    sort.Ints(nums)
    for i := 0; i < len(nums); i ++ {
        if 0 < i && nums[i] == nums[i - 1] {
            continue
        }
        
        k := len(nums) - 1
        for j := i + 1; j < len(nums); j ++ {
            if i + 1 < j && nums[j] == nums[j - 1] {
                continue
            }

            for j < k {
                sum := nums[i] + nums[j] + nums[k] 
                if sum > 0 {
                    k --
                } else if sum < 0 {
                    break
                } else {
                    if k + 1 < len(nums) && nums[k] == nums[k + 1] {
                        break
                    } 
                    ans = append(ans, []int{nums[i], nums[j], nums[k]})  
                    break                
                }
            }


            
        }
    }
    return ans
}