【Java】三数之和

三数之和

题目

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

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

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

示例 1:

输入: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] 。
注意,输出的顺序和三元组的顺序并不重要。

示例 2:

输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0

示例 3:

输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0

题解

暴力枚举

使用Set对结果进行去重,排序是为了去除结果集中顺序不同但是元素相同的数组,超时无疑

class Solution {
    
    
    public List<List<Integer>> threeSum(int[] nums) {
    
    
        if(nums==null || nums.length<3){
    
    
            return new ArrayList<>();
        }
        Set<List<Integer>> res=new HashSet<>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
    
    
            for (int j=i+1;j<nums.length;j++){
    
    
                for(int k=j+1;k<nums.length;k++){
    
    
                    if(nums[i]+nums[j]+nums[k]==0){
    
    
                        res.add(Arrays.asList(nums[i],nums[j],nums[k]));
                    }
                }
            }
        }
        return new ArrayList<>(res);
    }
}

双指针查找

三个元素求和,使用for循环遍历来进行固定第一个元素,然后剩余的元素来遍历求和,使得三个元素之和等于0,
当三个数的和等于0,表示匹配成功,然后左边的指针向右移动一位,右边的指针向左移动一位,继续匹配下一个满足条件的元素
当三个数的和小于0,那么左边的指针向右移动一位
当三个数的和大于0,那么右边的指针向左移动一位

class Solution {
    
    
    public List<List<Integer>> threeSum(int[] nums) {
    
    
        if(nums == null || nums.length<3){
    
    
            return new ArrayList<>();
        }
        Set<List<Integer>> res=new HashSet<>();
        Arrays.sort(nums);
        for(int i=0;i<nums.length;i++){
    
    
            int target =-nums[i];
            int left =i+1;
            int right=nums.length-1;
            while (left<right){
    
    
                int sum =nums[left]+nums[right];
                if(sum== target){
    
    
                    res.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    left++;
                    right--;
                }else if(sum<target){
    
    
                    left++;
                }else {
    
    
                    right--;
                }
            }
        }
        return new ArrayList<>(res);
    }
}

对双指针的简化

对于排序之后的一个数组,里面的元素会存在有相同的元素,而且处于相邻的位置,当遍历到相同元素的第一个的时候,

  1. 当满足条件的时候,两个指针都需要从两端向中间移动(在满足左指针小于右指针的情况下,左指针向右移动一位,右指针向左移动一位),
    1)当移动之后两个指针指向的元素的值没有发生变化,那么这个成立的条件是重复的,需要舍弃
    2)当移动之后两个指针指向的元素只有一个发生变化,那么这个情况一定不满足条件,指向相同元素的指针需要继续移动,直到指向不一样的元素
    3)当移动之后两个指针指向的元素的值都发生变化,那么进行求和,判断是否满足条件,不会发生重复现象
  2. 当不满足条件的时候,需要根据三个元素之和判断是哪个指针进行移动
    1)当三个元素之和小于零,左边指针在满足left<right的条件下向右移动,只有移动的指针指向不一样的元素的时候才会有可能满足条件,否则将一直在left<right的限制下移动寻找下一个不一样的元素
    2)当三个元素之和大于零,右边指针在满足left<right的条件下向左移动,只有移动的指针指向不一样的元素的时候才会有可能满足条件,否则将一直在left<right的限制下移动寻找下一个不一样的元素

综上所述,可以去掉响应的重复匹配的循环

class Solution {
    
    
    public List<List<Integer>> threeSum(int[] nums) {
    
    
        if(nums == null || nums.length<3){
    
    	//去除固定元素和左指针相同的情况,直接跳过一次循环
            return new ArrayList<>();
        }
        List<List<Integer>> res=new ArrayList<>();
        Arrays.sort(nums);
        for(int i=0;i<nums.length;i++){
    
    
            if (i > 0 && nums[i] == nums[i - 1]) continue;
            int target =-nums[i];
            int left =i+1;
            int right=nums.length-1;
            while (left<right){
    
    
                int sum =nums[left]+nums[right];
                if(sum== target){
    
    
                    res.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    while (left<right && nums[left]==nums[++left]);	//去除当重复满足条件的元素,和根本不可能满足条件的元素判断,下一行仍是如此
                    while (left<right && nums[right]==nums[--right]);
                }else if(sum<target){
    
    
                    //left++;
                    while (left<right && nums[left]==nums[++left]);//去除左指针向右移时候遇见相同元素的情况
                }else {
    
    
                    //right--;
                    while (left<right && nums[right]==nums[--right]);//去除右指针向左移时候遇见相同元素的情况
                }
            }
        }
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/m0_58121644/article/details/128616777