15. 三数之和

用吉他敲代码的恶犬

于 2021-07-09 19:02:04 发布

阅读量60

点赞数

分类专栏：双指针

本文链接：https://blog.csdn.net/qq_34241578/article/details/118612980

版权

双指针专栏收录该内容

6 篇文章 0 订阅

订阅专栏

本文介绍了如何利用排序和双指针算法解决LeetCode上的三数之和问题。首先对数组进行排序，然后通过一个外层循环确定三元组的第一个元素，接着使用双指针在剩余部分寻找和为0的其他两个元素。当找到满足条件的三元组时，需要避免重复添加，并根据和的正负调整指针位置。最终返回所有不重复的三元组。

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

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

题目描述

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
示例 2：

输入：nums = []
输出：[]
示例 3：

输入：nums = [0]
输出：[]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题过程

解题思路

双指针：先对数组进行从小到大的排序。然后for循环确定一个三元组的头并判断是否重复，重复则跳过这个头，在for循环里用双指针指向剩下两个三元组元素。在双指针遍历循环中判断三个元素的和的情况，等于0则左右指针都要判重，小于0则左指针要右移；大于0则右指针左移。

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        
        for(int i = 0; i < nums.length; i++){
            if(nums[i] > 0){//说明以nums[i]开头，包括i后面的都不行了（因为递增排序了）
                break;
            }
            if(i > 0 && nums[i] == nums[i - 1]){
                continue;
            }

            int L = i + 1;
            int R = nums.length - 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(R > L && nums[R] == nums[R - 1]){
                        R--;
                    }
                    L++;
                    R--;
                }else if(sum < 0){
                    L++;
                }else if(sum > 0){
                    R--;
                }
            }
        }

        return res;
    }
}