三数之和(java)

皮_客

已于 2023-07-04 18:40:24 修改

阅读量633

点赞数

分类专栏： Java 文章标签： python 开发语言 java leetcode 算法

于 2023-07-04 18:22:58 首次发布

本文链接：https://blog.csdn.net/qq_43593534/article/details/131540460

版权

Java 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

给定一个整数数组，该程序通过排序和双指针法找出所有和为0的不重复三元组。首先检查数组长度，然后对数组排序，用三层循环找到满足条件的三元组，并在过程中避免重复项。当和小于0时，增加第二个值；当和大于0时，减少第三个值。最终返回所有满足条件的三元组列表。

摘要由CSDN通过智能技术生成

给你一个整数数组 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] 。
注意，输出的顺序和三元组的顺序并不重要。

解析：将数组从小到大排序，使用三个标记量和两层遍历实现全部组合。

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();

        int len = nums.length;
        if(nums == null || len<3){  // 少于三个数不满足条件直接返回
            return res;
        }
        Arrays.sort(nums);  // 对数组进行排列
        for(int i = 0; i < len; i++){  // 最外层遍历，第一个值
            if(nums[i]>0){  // 特殊情况直接跳出
                break;
            }
            if(i > 0 && nums[i] == nums[i-1]) continue;  // 跳过重复值
            int l = i + 1;  // 第二个值
            int r = len - 1;  // 第三个值
            while(l<r){  // 第二层遍历
                int sum = nums[i] + nums[l] + nums[r];  // 题目判断
                if(sum == 0){  // 满足题目
                    res.add(Arrays.asList(nums[i], nums[l], nums[r]));  // 把结果存储
                    while(l < r && nums[l] == nums[l+1]){  // 跳过重复值
                        l++;  
                    }
                    while(l < r && nums[r] == nums[r-1]){  // 跳过重复值
                        r--;
                    }
                    l++;  
                    r--;
                }else if(sum < 0){  // 值小于0，总值需要增加，第二值右移
                    l++;
                }else if(sum > 0){  // 值大于0，总值需要减小，第三值左移
                    r--;
                }
            }
        }
        return res;  // 返回结果
    }
}