力扣【四数之和】

一、题目描述

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;
    }
};
相关推荐
ROBIN__dyc7 分钟前
表达式
算法
无限大.11 分钟前
c语言实例
c语言·数据结构·算法
六点半88815 分钟前
【C++】速通涉及 “vector” 的经典OJ编程题
开发语言·c++·算法·青少年编程·推荐算法
@haihi24 分钟前
冒泡排序,插入排序,快速排序,选择排序
数据结构·算法·排序算法
quaer27 分钟前
Open-Sora全面开源?
开发语言·算法·机器学习·matlab·矩阵
Hello.Reader38 分钟前
TopK算法在大数据重复数据分析中的应用与挑战
大数据·算法·数据分析
coduck_S12004zbj41 分钟前
csp-j模拟五补题报告
c++·算法·图论
丢掉幻想准备斗争43 分钟前
数据结构(栈和队列的实现)
数据结构
洛临_1 小时前
【C语言】基础篇
c语言·算法
_.Switch1 小时前
Python机器学习模型的部署与维护:版本管理、监控与更新策略
开发语言·人工智能·python·算法·机器学习