LeetCode18-四数之和

LeetCode18-四数之和

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

一、思路

之前写过三数之和，所以，四数之和的写法思路与三数之和一样，都是分而治之

（一）分而治之

先排序，再确定一个数后，就将这个问题转为三数之和了。

C++代码：

class Solution {
public:
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
		int i, j, m, n, sum;
		set<vector<int>> vector_nums;
		if (nums.size() < 4)
			return vector<vector<int>>(vector_nums.begin(),vector_nums.end());

		sort(nums.begin(), nums.end());

		for (i = 0; i < nums.size() - 3; i++) {
			if (i > 0 && nums[i] == nums[i - 1])
				continue;
			for (j = i + 1; j < nums.size() - 2; j++) {
				if (j > (i + 1) && nums[j] == nums[j - 1])
					continue;
				m = j + 1;
				n = nums.size() - 1;
				while (m < n) {
					sum = nums[i] + nums[j] + nums[m] + nums[n];
					if (sum == target)
						vector_nums.insert({ nums[i],nums[j],nums[m],nums[n] });
					if (sum > target)
						n--;
					else
						m++;
				}
			}
		}

		return vector<vector<int>>(vector_nums.begin(), vector_nums.end());
	}
};

执行效率：

不太好，和排名靠前的有不小的差距，然后我点开一个看了下，思路大致一样，没什么区别，不知道为什么差了这么多