【附JS、Python、C++题解】Leetcode面试150题(9)——三数之和

一、题目​​​​​

15. 三数之和

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足:
i!=ji!=kj!= k ,同时还满足:nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。

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

二、思路

  1. 我们要返回的是"所有和为0且不重复的三元组",这是一个数组类型,数组里的每一个元素都是三元组;

  2. 要有三个用于遍历的指针;

  3. 判断条件就两个:

  • i!=ji!=kj!= k
  • nums[i] + nums[j] + nums[k] == 0、
  1. 如果直接遍历,重复次数太多了,如何解决?

【联想到"两数之和Ⅱ"的那道题,因为有了一个"非严格递增"的顺序条件,我们得以简化遍历的过程;在这里也可以借鉴这个思路------创造一个顺序出来】

该题题解见如下文章:
【附JS、Python、C++题解】Leetcode面试150题(7)-CSDN博客

三、代码

① JavaScript

javascript 复制代码
function threeNums(nums){
    nums.sort((a,b)->a-b);
    let res = [];
    let l = nums.length;
    for(let i = 0; i<l-2; i++){
        if(i>0 && nums[i]===nums[i-1]){
            continue;
        }
        let j = i+1;
        let k = l-1;
        while(j<k){
            const sum = nums[i] + nums[j] + nums[k];
            if(sum === 0){
                res.push([nums[i], nums[j], nums[k]]);
                while(j<k && nums[j] === nums[j+1]){
                    j++;
                }
                while(j<k && nums[k] === nums[k-1]){
                    k--;
                }
                j++;
                i--;
            }else if(sum<0){
                j++;
            }
            else{
                K--;
            }
            
        }
    }
    return res;
}

② Python

python 复制代码
def three_sum(nums):
    nums.sort()  # 先对数组进行排序
    res = []
    length = len(nums)

    for i in range(length - 2):
        if i > 0 and nums[i] == nums[i - 1]:
            continue

        j, k = i + 1, length - 1

        while j < k:
            total = nums[i] + nums[j] + nums[k]
            if total == 0:
                res.append([nums[i], nums[j], nums[k]])
                # 避免重复计算
                while j < k and nums[j] == nums[j + 1]:
                    j += 1
                while j < k and nums[k] == nums[k - 1]:
                    k -= 1
                j += 1
                k -= 1
            elif total < 0:
                j += 1
            else:
                k -= 1

    return res

③ C++

cpp 复制代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

vector<vector<int>> threeSum(vector<int>& nums) {
    vector<vector<int>> res;
    int length = nums.size();

    // 先对数组进行排序
    sort(nums.begin(), nums.end());

    for (int i = 0; i < length - 2; ++i) {
        if (i > 0 && nums[i] == nums[i - 1]) {
            continue;
        }

        int j = i + 1;
        int k = length - 1;

        while (j < k) {
            int total = nums[i] + nums[j] + nums[k];
            if (total == 0) {
                res.push_back({nums[i], nums[j], nums[k]});
                // 避免重复计算
                while (j < k && nums[j] == nums[j + 1]) {
                    ++j;
                }
                while (j < k && nums[k] == nums[k - 1]) {
                    --k;
                }
                ++j;
                --k;
            } else if (total < 0) {
                ++j;
            } else {
                --k;
            }
        }
    }

    return res;
}

四、反思

这道题自己做的时候并没有先进行排序,导致重复的次数很多。下次遇到遍历很复杂的问题,要先进行处理!!

相关推荐
小南家的青蛙36 分钟前
LeetCode第51题 - N 皇后
算法·leetcode·职场和发展
Yupureki1 小时前
从零开始的C++学习生活 2:类和对象(上)
c语言·开发语言·c++·学习·visual studio
南屿im1 小时前
把代码变成“可改的树”:一文读懂前端 AST 的原理与实战
前端·javascript
mxd018481 小时前
最常用的js加解密之RSA-SHA256 加密算法简介与 jsjiami 的结合使用指南
开发语言·javascript·ecmascript
字节高级特工2 小时前
网络协议分层与Socket编程详解
linux·服务器·开发语言·网络·c++·人工智能·php
FS_tar3 小时前
高斯消元矩阵
c++·算法·矩阵
摸着石头过河的石头3 小时前
从零开始玩转前端:一站式掌握Web开发基础知识
前端·javascript
闻缺陷则喜何志丹4 小时前
【贪心之临项交换】P8732 [蓝桥杯 2020 国 ABC]|普及
c++·算法·蓝桥杯·贪心·洛谷
charlie1145141914 小时前
理解C++20的革命特性——协程支持1
c++·学习·c++20·协程·语言特性·调度·现代c++
而后笑面对4 小时前
关于力扣2025.10.4每日 11.盛最多雨水的容器
算法·leetcode·职场和发展