《leetcode-go》数组中相加和为0的三元组

给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。

注意:

  1. 三元组(a、b、c)中的元素必须按非降序排列。(即a≤b≤c)
  2. 解集中不能包含重复的三元组。
例如,给定的数组 S = {-10 0 10 20 -10 -40},解集为(-10, -10, 20),(-10, 0, 10) 
0 <= S.length <= 1000

思路:指定一个值,然后找另外两个值,找另外两个值就是头尾相向而行的方式,很经典

package main
import "sort"
/**
 * 
 * @param num int整型一维数组 
 * @return int整型二维数组
*/
func threeSum( num []int ) [][]int {
    // write code here
    if len(num) < 3 {
		return nil
	}
	sort.Ints(num)
	idx := 0
	res := make([][]int, 0)
	for i := 0; i < len(num)-2; i++ {
		if i > 0 && num[i] == num[i-1] {
			continue
		}
		idx = i
		start := i + 1
		end := len(num) - 1
		for start < end {
			if num[idx]+num[start]+num[end] == 0 {
				res = append(res, []int{num[idx], num[start], num[end]})
				start++
				for start < len(num)-1 && num[start] == num[start-1] {
					start++
				}
			}
			if num[idx]+num[start]+num[end] > 0 {
				end--
				for end < len(num)-1 && end >= 0 && num[end] == num[end+1] {
					end--
				}
            } else {
                start++
				for start < len(num)-1 && num[start] == num[start-1] {
					start++
				}
            }
		}
	}
	return res
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值