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, 0, 2]
]

解题思路:

这里为了避免重复项,我们使用了STL中的set,其特点是不能有重复,如果新加入的数在set中原本就存在的话,插入操作就会失败,这样能很好的避免的重复项的存在。此题的O(n3)解法的思路跟三数之和基本没什么区别,就是多加了一层for循环,其他的都一样,代码如下:

C++参考答案:

复制代码
 1 class Solution {
 2 public:
 3     vector<vector<int>> fourSum(vector<int> &nums, int target) {
 4         set<vector<int>> res;
 5         sort(nums.begin(), nums.end());
 6         for (int i = 0; i < int(nums.size() - 3); ++i) {
 7             for (int j = i + 1; j < int(nums.size() - 2); ++j) {
 8                 if (j > i + 1 && nums[j] == nums[j - 1]) continue;
 9                 int left = j + 1, right = nums.size() - 1;
10                 while (left < right) {
11                     int sum = nums[i] + nums[j] + nums[left] + nums[right];
12                     if (sum == target) {
13                         vector<int> out{nums[i], nums[j], nums[left], nums[right]};
14                         res.insert(out);
15                         ++left; --right;
16                     } else if (sum < target) ++left;
17                     else --right;
18                 }
19             }
20         }
21         return vector<vector<int>>(res.begin(), res.end());
22     }
23 };
复制代码

 

posted @   Ariel_一只猫的旅行  阅读(207)  评论(0编辑  收藏  举报
编辑推荐:
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示