从零刷算法(8) 三数之和

71 阅读1分钟

「这是我参与2022首次更文挑战的第8天,活动详情查看:2022首次更文挑战」。

题目描述

给你一个包含 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]
输出:[]

提示:

0 <= nums.length <= 3000 -105 <= nums[i] <= 105

题目链接:[三数之和](leetcode-cn.com/problems/ro…)

思路介绍

声明返回的结果 判断长度条件 排序 遍历排序后的数组 判断当前数字大于0,直接跳出循环,返回结果 判断当前数字是不是等于前一个数字,是的话,进行下一次循环 当前数字的下一个数字为左指针,最后一个数字为右指针 左右指针双向遍历 计算当前数字、遍历的左指针数字以及右指针数字的和 判断和是不是为0 为0,记录结果;判断左右数字是不是和下一个数字相等,如果相等左指针++,右指针--; 左指针++,右指针--; 大于0,说明右边值大了,右指针--; 小于0,说明左边值小了,左指针++; 返回结果

拿这个nums数组来举例,首先将数组排序,然后有一层for循环,i从下标0的地方开始,同时定一个下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上。

依然还是在数组中找到 abc 使得a + b +c =0,我们这里相当于 a = nums[i] b = nums[left] c = nums[right]。

接下来如何移动left 和right呢, 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。

如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。

代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
​
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] > 0) {
                return result;
            }
​
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
​
            int left = i + 1;
            int right = nums.length - 1;
            while (right > left) {
                int sum = nums[i] + nums[left] + nums[right];
                if (sum > 0) {
                    right--;
                } else if (sum < 0) {
                    left++;
                } else {
                    result.add(Arrays.asList(nums[i], nums[left], nums[right]));
​
                    while (right > left && nums[right] == nums[right - 1]) right--;
                    while (right > left && nums[left] == nums[left + 1]) left++;
                    
                    right--; 
                    left++;
                }
            }
        }
        return result;
    }
}

运行结果

执行结果:通过

执行用时:19 ms

内存消耗:42.4 MB

\