Java LeetCode 15. 三数之和

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]

先对特殊情况进行处理,

if(nums==null||nums.length<3){
            return Collections.EMPTY_LIST;
        }

首先先把该数组进行排序,方便后面操作

然后创建一个将要返回的数组list

然后做一个循环处理,循环的次数为nums.length-2,因为除了i之外还至少需要预留两个数字的位置

循环中首先需要判断如果nums[i]如果大于0了,那么直接可以结束该循环了,因为三数相加必定大于0,
如果i上的元素和上一次循环的i上的元素相同,那么就可能产生同样的集合,为了去重也是需要跳过此次循环

之后定义三个变量,分别为循环i位置的a,i+1位置的b和最后一个位置c.

如果三个数相加等于0,那么把这三个数存到一个新的集合中,然后把这个集合添加到一开始定义的集合list中,之后进行最重要的地方去重,如果b后面,c前面得数和现在一样,那么就进行b++或者c–;

 while(b<c&&nums[b]==nums[b+1]){
                      b++;
                  }
                  while(b<c&&nums[c]==nums[c-1]){
                      c--;
                  }

这时候便可以b++,c–继续判断是否有加起来等于0的;

如果a+b+c位置上的元素加起来大于0,就c–把右边大的数减小
如果a+b+c位置上的元素加起来小于0,就b++把左边大的数增大

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        
        if(nums==null||nums.length<3){
            return Collections.EMPTY_LIST;
        }
        Arrays.sort(nums);
        List<List<Integer>> list = new ArrayList();
        for(int i=0;i<nums.length-2;i++)
            if(nums[i]>0){
                break;
            }
            if(i>0&&nums[i]==nums[i-1]){
                continue;
            }
            int a = nums[i];
            int b = i+1;
            int c = nums.length-1;
            while(b<c){
                
                if((a+nums[b]+nums[c])==0){
                  List temp = new ArrayList();
                  temp.add(a);
                  temp.add(nums[b]);
                  temp.add(nums[c]);
                  list.add(temp);
                  while(b<c&&nums[b]==nums[b+1]){
                      b++;
                  }
                  while(b<c&&nums[c]==nums[c-1]){
                      c--;
                  }
                  b++;
                  c--;
            }   else if((a+nums[b]+nums[c])<0){
                  b++;
            }   else{
                  c--;
            }
            }
        }
        return list;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值