LeetCode 18 c++ 四数之和

Walden1998

于 2019-04-01 20:04:25 发布

阅读量277

点赞数

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/See_the_world/article/details/88956031

版权

Leetcode 专栏收录该内容

35 篇文章

订阅专栏

本文介绍了一种解决四数之和问题的高效算法。通过使用排序和双指针技术来减少搜索空间，实现了时间复杂度为O(n^3)的解决方案。该算法避免了重复结果，并利用集合来存储唯一解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <algorithm>
#include <cmath>
#include <set>
#include <vector>
using namespace std;
class Solution {
public:
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
		set<vector<int>> res;
		if (nums.size() < 4){
			return vector<vector<int>>(res.begin(),res.end());
		}
		int l, r, sum;
		sort(nums.begin(), nums.end());
		for (int i = 0; i < nums.size()-3; i++){
			for (int j = i + 1; j < nums.size() - 2; j++){
				l = j + 1;
				r = nums.size() - 1;
				sum = target - nums[i] - nums[j];
				while (l < r){
					if (nums[l] + nums[r] == sum){
						vector<int> scheme = { nums[i], nums[j], nums[l], nums[r] };
						res.insert(scheme);
						while (l < r&&nums[l + 1] == nums[l]) ++l;
						while (l < r&&nums[r - 1] == nums[r]) --r;
						++l;
						--r;
					}
					else if (nums[l] + nums[r] < sum) ++l;
					else --r;
				}
			}
		}
		return vector<vector<int>>(res.begin(), res.end());
	}
};
int main(){
	Solution s;
	vector<int> b = {1,0,-1,-2,0,2};
	int c = 1;
	vector<vector<int>> a;
	a = s.fourSum(b, 0);
	for (int i = 0; i < a.size(); i++){
		for (int j = 0; j < a[i].size(); j++){
			cout << a[i][j] << " ";
		}
		cout << endl;
	}
	return 0;
}