ThinkChat🤖让你学习和工作更高效,注册即送10W Token,即刻开启你的AI之旅 广告
## 一、题目描述 ![](https://box.kancloud.cn/2016-01-05_568bb5e83b2cf.jpg) ## 二、解题技巧 这道题从表面上看与3Sum极其相似,事实上确实可以使用相同的思维和方法,只不过这样做的话,时间复杂度为O(n^3),空间复杂度为O(1)将超时。 这道题也可以在排序之后先计算后面两个数的和,将其方法一个哈希表中,由于可能存在不同的两个数的和为相同值,因此,可以考虑将和为相同的值放在一个链表中,然后将变量头放在哈希表中。然后再按照3Sum的思路,不过第三个数在这里变成了第三个和第四个数的和,通过哈希表可以方便地找到和为固定值的数的链表,就可以找到符合条件的四个数。这种方法的时间复杂度为O(n^2),空间复杂度也为O(n^2)。 ## 三、实例代码 ~~~ // 该算法使用3Sum的思路来实现 #include <iostream> #include <algorithm> #include <vector> using namespace std; class solution { public: vector<vector<int> > fourSum(vector<int> &num, int target) { vector<vector<int> > result; int Size = num.size(); if (Size < 4) return result; // 对输入的数据进行排序 sort(num.begin(), num.end()); for (int first = 0; first < (Size - 3); first++) { if ((first != 0) && (num[first] == num[first - 1])) { continue; } int first_num = num[first]; for (int second = first + 1; second < (Size - 2); second++) { if ((second != first + 1) && (num[second] == num[second - 1])) { continue; } int second_num = num[second]; int third = second + 1; int forth = Size - 1; // 对第三和第四个数进行左右夹逼 while (third < forth) { int third_num = num[third]; int forth_num = num[forth]; int Sum = first_num + second_num + third_num + forth_num; if (Sum == target) { // 存放一维数据 vector<int> temp; temp.push_back(first_num); temp.push_back(second_num); temp.push_back(third_num); temp.push_back(forth_num); // 将组合存入容器result中 result.push_back(temp); third++; while ((third < Size - 1) && (num[third] == num[third - 1])) { third++; } forth--; while ((forth > second) && (num[forth] == num[forth + 1])) { forth--; } } if (Sum > target) { forth--; while ((forth > second) && (num[forth] == num[forth + 1])) { forth--; } } if (Sum < target) { third++; while ((third < Size) && (num[third] == num[third - 1])) { third++; } } } } } return result; } }; ~~~ 测试: ~~~ #include "solution.h" using namespace std; int main() { solution a; int size = 200; // 数据的个数 vector<int> num(size); // 用于存放数据的容器 vector<vector<int> > result; // 存放算法的运行结果 for (int i = 0; i < size; i++) num.push_back(rand() % 30 - 20); // 随机生成数据 int target = 0; result = a.fourSum(num, target); for (vector<vector<int> >::size_type j = 0; j < result.size(); j++) { for (vector<int>::size_type k = 0; k < result[j].size(); k++) { cout << result[j][k] << " "; } cout << endl; } getchar(); return 0; } ~~~ 一个输出结果: ![](https://box.kancloud.cn/2016-01-05_568bb5e864da4.jpg) ## 四、体会 这道题是3Sum的一种延伸,但需要转化一种思路以降低计算复杂度。如果仅仅按照3Sum的延伸来做这道题的话,算法的空间复杂度会达到O(n^3),但是空间复杂度为O(1)。可以换一种思路,在排序之后,先计算后面两个数的和,并用哈希表存储起来,然后就将问题转变为了3Ssum的问题,只不过计算出第三个数之后,还需要在哈希表中找到和满足条件的第三个数和第四个数。