漫画:经典鹅厂面试题(4sum - nSum)


今天是小浩算法“365刷题计划”第79天。为大家讲解三数之和的进阶版本 - 四数之和。

建议先回顾一下该题前两个版本:

漫画:两数之和

漫画:经典鹅厂面试题(2Sum,3Sum,4Sum)

01

PART

四数之和

本题是 三数之和 的进阶版本。

第18题:给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:

答案中不可以包含重复的四元组。

示例:

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:

[

  [-1,  0, 0, 1],

  [-2, -1, 1, 2],

  [-2,  0, 0, 2]

]

02

PART

题目分析

对于四数之和,其实和三数之和差别不大,仍然采用双指针来求解即可。

之前有人问我“三数之和我知道通过固定一个元素,使用双指针遍历进行求解,但是四数之和我就搞不清楚如何固定元素了怎么办”其实这个本质上还是一样的,一起来看下:

假若数组为 [1, 0, -1, 0, -2, 2],这样:

我们第一步肯定还是先排序(这个在三数之和里也讲了,排序的目的有两个:1、方便跳过重复元素 2、可以利用排序的特性,直接过滤掉重复计算

排完序的数组长这样:

然后下来就是设置我们需要的四个值了:(待固定的两个值,以及双指针)

其实固定的方法还是一样,因为我们已经固定了 i,只需要对 i+1 就可以得到我们待固定的 j。放在代码里,其实就是多了一个嵌套循环。然后差不多可以给出下面的代码代码结构:

 1for (int i = 0; i < n - 3; i++) {
 2    //特殊处理-1
 3    if (i > 0 && nums[i] == nums[i - 1]) continue;
 4    if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;
 5    if (nums[i] + nums[n - 1] + nums[n - 2] + nums[n - 3] < target) continue;
 6    for (int j = i + 1; j < n - 2; j++) {
 7        //特殊处理-2
 8        if (j - i > 1 && nums[j] == nums[j - 1]) continue;
 9        if (nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) break;
10        if (nums[i] + nums[j] + nums[n - 1] + nums[n - 2] < target) continue;
11        int left = j + 1;
12        int right = n - 1;
13        while (left < right) {
14            //双指针循环查找
15        }
16    }

当然,这个结构里,除了对四个值的定义,我们还加了两段特殊处理的逻辑。分别是 3-5 行 和 8-10 行。这是什么意思呢:

  • 第 3 行和第 8 行分别处理了 i 和 j 重复值的情况。

  • 第 4-5 和 9-10 就是我们上面说的,利用排序的特性,直接过滤掉重复计算了。

因为题目中已经要求了,不允许出现重复值,所以第3行和第8行肯定是不可以去掉的,但是题目中没说非得利用排序的特性对不?那一会儿我们就可以尝试把 4-5 和 9-10 去掉,看看会不会报错,这个一会儿再说。

剩下的逻辑就是在双指针循环体内查找另外两个元素了。自然这里我们也得处理重复值(第7行和第8行),到这里细心的读者会发现,其实和三数之和已经差不多了!

 1//JAVA
 2while (left < right) {
 3    int tmp = nums[i] + nums[j] + nums[left] + nums[right];
 4    if (tmp == target) {
 5        List<Integer> tmpList = new LinkedList<>(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
 6        res.add(tmpList);
 7        while (left < right && nums[left] == nums[left + 1]) left += 1;
 8        while (left < right && nums[right] == nums[right - 1]) right -= 1;
 9        left += 1;
10        right -= 1;
11    } else if (tmp > target) right -= 1;
12    else left += 1;
13}

剩下的自然就是拼装代码了:

 1//JAVA
 2class Solution {
 3    public List<List<Integer>> fourSum(int[] nums, int target) {
 4        List<List<Integer>> res = new LinkedList<>();
 5        Arrays.sort(nums);
 6        int n = nums.length;
 7        for (int i = 0; i < n - 3; i++) {
 8            if (i > 0 && nums[i] == nums[i - 1]) continue;
 9            if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;
10            if (nums[i] + nums[n - 1] + nums[n - 2] + nums[n - 3] < target) continue;
11            for (int j = i + 1; j < n - 2; j++) {
12                if (j - i > 1 && nums[j] == nums[j - 1]) continue;
13                if (nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) break;
14                if (nums[i] + nums[j] + nums[n - 1] + nums[n - 2] < target) continue;
15                int left = j + 1;
16                int right = n - 1;
17                while (left < right) {
18                    int tmp = nums[i] + nums[j] + nums[left] + nums[right];
19                    if (tmp == target) {
20                        List<Integer> tmpList = new LinkedList<>(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
21                        res.add(tmpList);
22                        while (left < right && nums[left] == nums[left + 1]) left += 1;
23                        while (left < right && nums[right] == nums[right - 1]) right -= 1;
24                        left += 1;
25                        right -= 1;
26                    } else if (tmp > target) right -= 1;
27                    else left += 1;
28                }
29            }
30
31        }
32        return res; 
33    }
34}

然后跑一跑,发现还不错:

然后我们再尝试把刚才说的“避免重复计算”的代码去掉:

 1//JAVA
 2class Solution {
 3    public List<List<Integer>> fourSum(int[] nums, int target) {
 4        List<List<Integer>> res = new LinkedList<>();
 5        Arrays.sort(nums);
 6        int n = nums.length;
 7        for (int i = 0; i < n - 3; i++) {
 8            if (i > 0 && nums[i] == nums[i - 1]) continue;
 9            for (int j = i + 1; j < n - 2; j++) {
10                if (j - i > 1 && nums[j] == nums[j - 1]) continue;
11                int left = j + 1;
12                int right = n - 1;
13                while (left < right) {
14                    int tmp = nums[i] + nums[j] + nums[left] + nums[right];
15                    if (tmp == target) {
16                        List<Integer> tmpList = new LinkedList<>(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
17                        res.add(tmpList);
18                        while (left < right && nums[left] == nums[left + 1]) left += 1;
19                        while (left < right && nums[right] == nums[right - 1]) right -= 1;
20                        left += 1;
21                        right -= 1;
22                    } else if (tmp > target) right -= 1;
23                    else left += 1;
24                }
25            }
26        }
27        return res; 
28    }
29}

发现也是不会报错的,但是性能就大打折扣了:

郑重申明(读我的文章必看):

  • 本系列所有教程都不会用到复杂的语言特性,大家无须担心没有学过相关语法,算法思想才是最重要的!

  • 作为学术文章,虽然风格可以风趣,但严谨,我是认真的。本文所有代码均在leetcode进行过测试运行。

03

PART

结尾

其实四数之和已经比较麻烦了,但是如果面试的时候对方问到N数之和,这个时候其实也不用着急。一般来讲,N数之和 是不会让你写的。只需要把上面的代码转化成递归的形式,答出来通过递归每次固定N-2个数就可以了。

2sum,3sum,4sum 系列篇 到这里就结束了。这里边 2sum 和 3sum 还是非常建议大家去练习一下的。算法这个东西,思想是一回事,动手是另一回事。长路漫漫磕磕绊绊,难受就是提高。

如果想看其他面试题相关内容的,可以看:

漫画:知乎面试题(旋转数组最小值Ⅰ - 基础版)

漫画:知乎面试题(旋转数组最小值Ⅱ - 进阶版)

漫画:美团面试题(TOPK:求第K个最大的元素)

漫画:腾讯面试题(面试官问我会不会修供暖器,我说没问题)

漫画:位运算技巧整理汇总+一道被嫌弃的题目

如果你问我对学习算法有什么建议,这篇文章是必看的:

漫画:呕心泣血算法指导篇(真正的干货,怒怼那些说算法没用的人)

 小浩算法,每日

关注领取《图解算法》高清版

进群的小伙伴请加右侧私人微信(备注:进群)

-----抽奖分隔线----

???????????? 扫码回复“红包”参与抽奖

金额:100元 | 个数:20个 | 中奖率:20%

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值