力扣【四数之和】

一、题目描述

18. 四数之和

给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复 的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):

  • 0 <= a, b, c, d < n
  • abcd 互不相同
  • nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按 任意顺序 返回答案 。

二、题目解析

算法思想:排序+双指针

1、依次固定一个数a;

2、在a后面的区间内,用"三数之和"找到三个数,使用这三个数的和等于target - a即可

同理,对于三数之和的算法:

1、依次固定一个数b;

2、在b后面的区间内,利用"双指针"找到两个数,是这两个数的和等于target - a - b

处理细节问题:

1、不重复

注意这里的去重和三数之和不同,这里需要多一组去重,在确定a时也是需要判断是否重复,其余去重操作和判断三数之和是一样。

2、不漏

与三数之和一样,在找到满足题目条件的一组元素之后,需要继续寻找。

注意:

这里的数据有溢出的风险,不开long long见祖宗~

三、原码

复制代码
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ret;
        //1、先排序
        sort(nums.begin(),nums.end());
        //2、利用双指针解决
        int n = nums.size();
        for(int i = 0;i<n;)
        {
            //下面是三数之和
            for(int j = i+1;j<n;)
            {
                //防止数据溢出,开long long
                long long target2 = (long long)target - nums[i] - nums[j];
                int left = j+1;
                int right = n-1;
                while(left < right)
                {
                    int sum = nums[left] + nums[right];
                    if(sum > target2) 
                        right--;
                    else if(sum < target2) 
                        left++;
                    else
                    {
                        ret.push_back({nums[i],nums[j],nums[right],nums[left]});
                        left++;
                        right--;
                        //去重left right
                        while(left < right && nums[left] == nums[left-1]) left++;
                        while(left < right && nums[right] == nums[right+1]) right--;
                    }
                }
                //去重j
                j++;
                while(j<n && nums[j] == nums[j-1]) j++;
            }
            //去重i
            i++;
            while(i<n && nums[i] == nums[i-1]) i++;
        }
        return ret;
    }
};
相关推荐
算AI9 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
我不会编程55510 小时前
Python Cookbook-5.1 对字典排序
开发语言·数据结构·python
owde11 小时前
顺序容器 -list双向链表
数据结构·c++·链表·list
第404块砖头11 小时前
分享宝藏之List转Markdown
数据结构·list
hyshhhh11 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
蒙奇D索大12 小时前
【数据结构】第六章启航:图论入门——从零掌握有向图、无向图与简单图
c语言·数据结构·考研·改行学it
A旧城以西12 小时前
数据结构(JAVA)单向,双向链表
java·开发语言·数据结构·学习·链表·intellij-idea·idea
杉之12 小时前
选择排序笔记
java·算法·排序算法
烂蜻蜓12 小时前
C 语言中的递归:概念、应用与实例解析
c语言·数据结构·算法
OYangxf12 小时前
图论----拓扑排序
算法·图论