3Sum 找出数组中的三元组,满足三元组之和为0

Given an array S of n integers, are there elements abc 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]
]
需要我们找出 sum = 0 的三元组。这题的难点是如何找出不重复的三元组(a, b, c)。

做法是先将array进行排序。

我们让a 为最小的,

接着让 b 为 a 的后一个,让c 为最大的哪一个,两端进行逼近。

如图中,排序后的S 为 [ -4, -1, -1, 0, ,1, 2]。

此时 a = -4, b = -1, c = 2。 sum = -3 . 所以b不够大,b再右走。

a 遍历完一遍数组后,就找完所有符合要求的了。

复杂度:

时间复杂度:O(n^2)

空间复杂度:O(1)

代码:

    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            int lo = i + 1;
            int hi = nums.length - 1;
            while (lo < hi) {
                if ((nums[i] + nums[lo] + nums[hi]) == 0) {
                    result.add(Arrays.asList(nums[i], nums[lo], nums[hi]));
                    while (lo + 1 <= hi && nums[lo + 1] == nums[lo]) {
                        lo++;
                    }
                    while (hi - 1 >= lo && nums[hi - 1] == nums[hi]) {
                        hi--;
                    }
                    lo++;
                    hi--;
                } else if ((nums[i] + nums[lo] + nums[hi]) > 0) {
                    hi--;
                } else {
                    lo++;
                }
            }
        }
        return result;
    }


  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值