面试常考的三数之和算法题,简明扼要带你拿下。

在这里插入图片描述

Leetcode题目链接

本题要求读者先掌握 两数之和 II - 输入有序数组 这道题,最好看我的 个人题解 以便于衔接本篇思路。

思路

先对数组排序,设一非递减的数组示例和初始三指针位置及名字如下所示。
15_2.png
固定 i i i,即可转换为寻找满足 n u m s [ l ] + n u m s [ r ] = − n u m s [ i ] nums[l] + nums[r] = -nums[i] nums[l]+nums[r]=nums[i] 的三元组,借用两数之和的思路即可。

还要注意因为不能包含重复的三元组,以下两个三元组只能取一个,而后我们再考虑其是否满足 n u m s [ l ] + n u m s [ r ] = − n u m s [ i ] nums[l] + nums[r] = -nums[i] nums[l]+nums[r]=nums[i]

15_3.png

所以在移动指针的时候,需要规避连续的重复元素,比如下方的移动 r r r,对于 i , l i, l i,l 同理。
15_4.png

题目中要求不重复,所以从小到大枚举 i i i l l l 的初值为 i + 1 i+1 i+1 r r r 的初值一直为 n u m s . s i z e − 1 nums.size - 1 nums.size1

下图为 i = 1 i = 1 i=1 的三指针初始位置。
Screenshot 2024-06-29 at 5.04.02 PM.png

代码

仅提供 java 代码,多语言代码块可看这里

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        // 待返回的三元组
        List<List<Integer>> triples = new ArrayList<>();

        for (int i = 0; i < nums.length; i++) {
            // 检测重复的 nums[i]
            if (i > 0 && nums[i] == nums[i - 1]) continue;

            int l = i + 1;
            int r = nums.length - 1;

            while (l < r) {
                // 检测重复的 nums[l]
                if (l > i + 1 && nums[l] == nums[l - 1]) {
                    l++;
                }
                // 检测重复的 nums[r]
                else if (r < nums.length - 1 && nums[r] == nums[r + 1]) {
                    r--;
                }
                // 均不重复再按照两数之和的思路
                // 等同于判断 nums[l] + nums[r] > -nums[i]
                else if (nums[i] + nums[l] + nums[r] > 0) {
                    r--;
                }
                // 等同于判断 nums[l] + nums[r] < -nums[i]
                else if (nums[i] + nums[l] + nums[r] < 0) {
                    l++;
                } else { // nums[l] + nums[r] + nums[i] == 0, 三元组符合,添加入结果
                    triples.add(Arrays.asList(nums[i], nums[l], nums[r]));
                    l++;
                    r--;
                }
            }
        }

        return triples;
    }
}

笔者代码中少了一层嵌套循环(存疑可看这里的详细对比),但复杂度不变。虽然耗时明显更慢,但在面试中是肯定能通过的。此外这种技巧在实际开发中也能用到,而且实际开发的内部操作繁多,相比之下,这种减少嵌套循环增加的耗时就可以忽略了。

复杂度

时间: O ( n 2 ) O(n^2) O(n2)
空间: Θ ( l o g   n ) \Theta(log\ n) Θ(log n),主要考虑排序,忽略返回值。

推广

这里有附个人题解的双指针题单

笔者著有 Leetbooks 《图论入门》《图论进阶》,且在陆续写其他 Leetbooks。致力于提升广大读者真正的算法水平,并兼顾不同阶段的读者,如本科阶段、硕士阶段、职场面试、算法竞赛。

  • 21
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值