前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >15. 三数之和

15. 三数之和

作者头像
chuckQu
发布2022-08-19 14:20:45
2270
发布2022-08-19 14:20:45
举报
文章被收录于专栏:前端F2E

15. 三数之和

力扣题目链接[1]

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

「示例 1:」

代码语言:javascript
复制
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]

「提示:」

  • 0 <= nums.length <= 3000
  • -10^5 <= nums[i] <= 10^5

思路:

本题是两数之和的升级版,但是思路和两数之和不尽相同。

本题采用排序+双指针的方式题解。首先确保数组是有序数组,然后固定一个元素,使用双指针分别指向固定元素的下一个元素和数组末尾元素,将三者元素相加进行判断。因为数组是有序的,所以和大于0时,左移右指针;和小于0时,右移左指针。等于0时,将结果放入结果数组。

由于题目要求不能包含重复的集合,因此需要去重处理。当遇到前后相同元素时,要直接跳过。

双指针

代码语言:javascript
复制
/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
    let length = nums.length; // 缓存数组长度
    if (!nums || length < 3) return []; // 提前返回
    let result = []; // 初始化结果数组
    let left = null; // 初始化左指针
    let right = null; // 初始化右指针
    nums.sort((a, b) => a - b); // 对数组进行排序
    for (let i = 0; i < length; i++) {
        if (nums[i] > 0) break; // 如果第一个数就大于0,那么后续相加肯定大于0,提前返回
        if (i > 0 && nums[i] === nums[i - 1]) continue; // 去重
        left = i + 1; // 左指针指向当前索引的下一个索引
        right = length - 1; // 右指针指向数组最后一个元素
        while(left < right) { // 循环终止条件是左指针大于等于右指针
            let num = nums[i] + nums[left] + nums[right]; // 三数之和
            if (!num) { // 和等于0
                result.push([nums[i], nums[left], nums[right]]); // 放入结果数组
                while(left < right && nums[left] === nums[left + 1]) left++; // 去重
                while(left < right && nums[right] === nums[right - 1]) right--; // 去重
                left++; // 结果中不包含重复的集合,因此移动左右指针
                right--;
            }
            else if (num < 0) left++; // 和大于0,右移左指针
            else if (num > 0) right--; // 和小于0,左移右指针
        }
    }
    return result;
};

参考资料

[1]

力扣题目链接: https://leetcode-cn.com/problems/3sum/

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2022-04-04,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 前端F2E 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 15. 三数之和
  • 双指针
    • 参考资料
    领券
    问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档