LeetCode数组中和为 0 的三个数

100 篇文章 2 订阅

剑指 Offer II 007. 数组中和为 0 的三个数

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。

来源:Leetcode

一开始想到的就是暴力解法,但是,当题目给出的数组是【0,0,0,0,0,…0】的时候,先进行三重循环,可以得到n的三次方组答案,然后再进行去重操作,得到的答案是null,但是这个过程就耗费了很多时间。

最好的办法就是不重复。
在这里插入图片描述

也就是保证i<j<k

先对数组进行排序;
在第一次循环中,如果取到了和之前的i一样的值,就跳过,也就是continue;

第二次循环里面,对于j的操作和上面一样,但是要注意,当i和j是同样的值时,这种情况当然也是成立的,应该考虑,只需要加一个判断条件,j>i+1即可,就可以将i,j对应元素值相同的情况考虑进去。
(例如,i和j都是-1的时候,在最终的答案中也是一种情况)

在第二重循环里面还可以进行改进,那就是j和k分别从两个方向取值,因为i+j+k=0,所以j+k=-i,由于已经对数组进行排序,所以,当j+k>-i的时候,随着k–,才有可能满足条件

但是k不能无限减下去,当j==k的时候如果还没出现答案,那就说明此次遍历无效,就break

最后,在找到答案的同时新建list集合,先存储一个答案,将其加入到ans中,每出现一个答案就加入一个。

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        int len=nums.length;
        Arrays.sort(nums);
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
        
        for(int i=0;i<len;i++)
        {
            if(i>0 && nums[i] == nums[i-1]){
                continue;
            }
            int flag=-nums[i];
            int k=len-1;
            for(int j=i+1;j<len;j++)
            {
                if(j>i+1&&nums[j]==nums[j-1])
                {
                    continue;
                }
                while(nums[j]+nums[k]>flag&&j<k)
                {
                    k--;
                }
                if(j==k)
                {
                    break;
                }
                if(nums[j]+nums[k]==flag)
                {
                    List<Integer> res = new ArrayList<Integer>();
                    res.add(nums[i]);
                    res.add(nums[j]);
                    res.add(nums[k]);
                    ans.add(res);
                }

            }
        }
        return ans;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值