每天AC系列(四):四数之和

1 题目

Leetcode第18题,给定一个数组与一个target,找出数组中的四个数之和为target的不重复的所有四个数.

2 暴力

List<List<Integer>> result = new ArrayList<>();
if (nums.length == 4 && nums[0] + nums[1] + nums[2] + nums[3] == target)
    result.add(Arrays.asList(nums[0], nums[1], nums[2],nums[3]));
else if (nums.length > 4)
{
    Arrays.sort(nums);
    Set<List<Integer>> resultSet = new HashSet<>();
    for(int i=0;i<nums.length-3;++i)
    {
        for(int j=i+1;j<nums.length-2;++j)
        {
            for(int k=j+1;k<nums.length-1;++k)
            {
                for(int m=k+1;m<nums.length;++m)
                {
                    if(nums[i]+nums[j]+nums[k]+nums[m] == target)
                        resultSet.add(Arrays.asList(nums[i],nums[j],nums[k],nums[m]));
                }
            }
        }
    }
    result.addAll(resultSet);
    Collections.sort(result,(t1,t2)->
    {
        if(t1.get(0) > t2.get(0))
            return 1;
        if (t1.get(0) < t2.get(0))
            return -1;
        if (t1.get(1) > t2.get(1))
            return 1;
        if (t1.get(1) < t2.get(1))
            return -1;
        if (t1.get(2) > t2.get(2))
            return 1;
        if (t1.get(2) < t2.get(2))
            return -1;
        if (t1.get(3) > t2.get(3))
            return 1;
        if (t1.get(3) < t2.get(3))
            return -1;
        return 0;
    });
}
return result;

判断长度,然后排序,直接上四个for,然后...

好! 惨败.

3 优化

3.1 去掉结果排序

首先最后的排序是不必要的,也就是后面的

 Collections.sort(result,(t1,t2)->
{
    if(t1.get(0) > t2.get(0))
        return 1;
    if (t1.get(0) < t2.get(0))
        return -1;
    if (t1.get(1) > t2.get(1))
        return 1;
    if (t1.get(1) < t2.get(1))
        return -1;
    if (t1.get(2) > t2.get(2))
        return 1;
    if (t1.get(2) < t2.get(2))
        return -1;
    if (t1.get(3) > t2.get(3))
        return 1;
    if (t1.get(3) < t2.get(3))
        return -1;
    return 0;
});

对结果进行排序不必要,虽然会在测试时与答案有差别,但是提交的话不需要排序.

3.2 stream去重

之前的操作用的是HashSet进行去重,有一个符合的四元组就直接添加进集合中,现在采用了stream+distinct去重:

return result.stream().distinct().collect(Collectors.toList());

3.3 双指针+最大最小剪枝

可以利用类似三数之和的思想,固定一个数,双指针分别指向两端的两个数,这里的话,四个数,选择固定两个数,计算它们的和并把它们看作一个数,即可利用双指针.

for(int i=0;i<nums.length-3;++i)
{
     for(int j=i+1;j<nums.length-2;++j)
     {
         int m = nums[i] + nums[j];
         int left = j+1;
         int right = nums.length-1;
         while(left < right)
         {
             int temp = m + nums[left] + nums[right];
             if(temp == target)
             {
                 result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                 --right;
                 ++left;
             }
             else if(temp > target)
                 --right;
             else
                 ++left;
         }
     }
 }

m为固定的数,left与right就是双指针,根据"三数"之和判断与目标值的大小移动双指针.
最小剪枝就是首先计算"三数"的最小值,若大于目标值就可以跳过,最大剪枝就是计算"三数"的最大值,若小于目标值则跳过,进入下一个循环:

int m = nums[i] + nums[j];
int left = j+1;
int right = nums.length-1;
if(m + nums[left] + nums[left+1] > target)
    continue;
if (m + nums[right-1] + nums[right] < target)
    continue;

3.4 提交


呃...好了那么一点点吧.

4 来来来再快一点

4.1 初始判断

首先,初始的判断可以再简单一点,如果数组为空或长度小于4,直接返回.

List<List<Integer>> result = new ArrayList<>();
if (nums == null && nums.length < 4)
    return result;

4.2 一次不够,就再剪几次

上面的算法中,只是在两层for里面进行了一次最大最小剪枝,可以在没进入for之前剪一次:

Arrays.sort(nums);
int len = nums.length;
if(
    nums[0] + nums[1] + nums[2] + nums[3] >  target
    ||
    nums[len-4] + nums[len-3] + nums[len-2] + nums[len-1] < target
)
    return result;
for(int i=0;i<len-3;++i)

注意要先排序,然后直接判断整个数组的最大最小值并与target判断.
然后在进入第一层for之后再剪一次:

for(int i=0;i<len-3;++i)
{
    if(nums[i] + nums[i+1] + nums[i+2] + nums[i+3] > target)
        break;
    if(nums[i] + nums[len-3] + nums[len-2] + nums[len-1] < target)
        continue;
    for(int j=i+1;j<len-2;++j)
}

因为数组是升序排序的,因此,"最左边"四个数肯定是最小值,若这个最小值大于target,可以直接break了,但是,最右边三个数与nums[i]相加不一定为最大值,因此判断之后若小于target只能continue.

4.3 去重

4.3.1 双指针去重

首先,在双指针的循环中,若发现了有四个数符合条件,可以尝试多次移动指针:

result.add(Arrays.asList(nums[i], nums[j], nums[left++], nums[right--]));
while(left < right && nums[left] == nums[left-1])
    ++left;
while(left < right && nums[right] == nums[right+1])
    --right;

因为值一样的可以一次性移动指针,不需要再次进行和的判断.
呃,可以尝试提交了.

咦,不对啊,做了这么多,没快多少啊...
为啥呢...
...

4.3.2 外循环去重

找了很久,发现是这里的原因:

return result.stream().distinct().collect(Collectors.toList());

这里去重的话,用是用的很舒服,一个stream(),一个distinct()就好了,问题是...还是很慢啊!!!
所以呢,需要手动去重,出现重复的原因就是数组中有重复的数,比如:

[1,1,1,1,2,2,2,2],target=6

顺序判断时,会好几个

[1,1,2,2]

因此,对于重复的数,进行跳过处理,在第一层for中,对重复过的进行跳过:

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

其次,在第二层for中,也对重复过的进行跳过:

for(int j=i+1;j<len-2;++j)
    if(j > i+1 && nums[j] == nums[j-1]) continue;

这样的话,例如对于上面的(不同的1用字母区分)

[1(a),1(b),1(c),1(d),2,2,2,2]

一开始是a处的1与b处的1,然后到了第二层循环,因为此时j=i+1,指向b处的1,因此不会跳过1,会进入双指针循环,第二次j指向c处的1,出现重复,j不断跳过直到j指向2.然后2结束后,到了i这层循环,因为1出现过,i不断跳过直到i指向2.

没错,说了这么多,去重不需要什么HashSet,不需要什么stream,只需两行:

if(i>0 && nums[i] == nums[i-1]) continue;
if(j>i + 1 && nums[j] == nums[j-1]) continue;

4.4 提交


尽力了.

5 源码

github

码云

原文地址：https://www.cnblogs.com/Blueeeeeeee/p/12234098.html

时间： 2024-11-14 12:20:11

每天AC系列(四):四数之和的相关文章

每天AC系列(一):三数之和

1 题目 LeetCode第15题,难度中等,题目描述: 给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组. 注意:答案中不可以包含重复的三元组. 2 解法 什么也不管先来个O(n3): for(int i=0;i<nums.length;++i) { for(int j=i+1;j<nums.length;++j) { for(int k=j+1;k<nums.length;

[leetcode数组系列]2三数之和

前言 秋招的结束,面试了大大小小的公司,最大的问题在于算法上.所以打算坚持在leetcode打卡,看看到底能不能行,如果你想见证,那我来开车,你坐稳,一起走向更好的远方. 在学习今天内容之前,先学习上一篇的两数之和会更好哟 leetcode两数之和求解 一 题目 给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满

每天AC系列(十三):两数相除

1 题目 LeetCode第29题,计算两数相除的商,不允许使用乘法,除法,求模运算符. 2 减法 首先判断结果是否需要加上负号,将商置为0后,被除数不断减去除数,同时商自增.最后根据是否有负号返回相应的商. boolean negative = true; if((dividend > 0 && divisor > 0) || (dividend < 0 && divisor < 0)) negative = false; dividend = d

LeetCode 18. 4Sum （四数之和）

Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target. Note: The solution set must not contain duplicate quadruplets. For exampl

[LeetCode] 454. 4Sum II 四数之和II

Given four lists A, B, C, D of integer values, compute how many tuples (i, j, k, l) there are such that A[i] + B[j] + C[k] + D[l] is zero. To make problem a bit easier, all A, B, C, D have same length of N where 0 ≤ N ≤ 500. All integers are in the r

LeetCode（18）：四数之和

Medium! 题目描述: 给定一个包含 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

Leetcode（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] ] 思路:

LeetCode 18. 四数之和（4Sum）

题目描述 给定一个包含 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] ]

【Leetcode】【4Sum】【四数之和】【C++】

题目:给定一个整数数组nums和一个目标值target,判断是否存在四个数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] ] 说明: 注意找出的四元组不重合,即没有相同的四元组:且四元组按从小到大的顺序输出. 思路1:对nums数组排序,然后两个for循