18. 四数之和

时间:2019-08-25
本文章向大家介绍18. 四数之和,主要包括18. 四数之和使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

给定一个包含 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]
]

/*
解题思路:
解法的思路跟 3Sum 基本没啥区别,就是多加了一层 for 循环
注意去重处理
*/
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
	vector<int> ns;
	vector<vector<int>> ans;
	bool isok(vector<vector<int>> ans, int i, int j, int k)
	{
		for (int u = 0; u < ans.size(); u++)
		{
			if (i == ans[u][0] && j == ans[u][1] && k == ans[u][2])
				return false;
		}
		return true;
	}
	vector<vector<int>> fourSum(vector<int>& nums, int target) 
	{
		sort(nums.begin(), nums.end());
		int N = nums.size();
		for (int i = 0; i < N - 3; i++)
		for (int j = i + 1; j<N - 2; j++)
		for (int k = j + 1; k <= N - 1; k++)
		for (int l = k + 1; l<N; l++)
		if (nums[i] + nums[j] + nums[k] + nums[l] == target)
		{
			if (isok(ans, nums[i], nums[j], nums[k]))
			{
				ns.push_back(nums[i]);   
				ns.push_back(nums[j]);
				ns.push_back(nums[k]);   
				ns.push_back(nums[l]);
				ans.push_back(ns);       
				ns.clear();
			}
		}
		return ans;
	}
};
int main(){
	vector<int>nums;
	int target;
	cin >> target;
	int a[1000];
	int i = 0;
	int x;
	while (cin >> a[i])
	{
		x = cin.get();
		if (x == '\n')
			break;
		nums.push_back(a[i]);
		++i;
	}
	vector<vector<int>>A = Solution().fourSum(nums, target);
	for (int i = 0; i < A.size(); i++)
		for (int j = 0; j < A[0].size(); j++)
			cout << A[i][j] << " ";
		    cout << endl;
	system("pause");
	return 0;
}

  

原文地址:https://www.cnblogs.com/277223178dudu/p/11407139.html