数组-四数之和-中等

描述
给一个包含n个数的整数数组S,在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。
四元组(a, b, c, d)中,需要满足a <= b <= c <= d
答案中不可以包含重复的四元组。

样例
例如,对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为:
(-1, 0, 0, 1)
(-2, -1, 1, 2)

(-2, 0, 0, 2)

题目链接

分析

借助三数求和的思想,除了考虑重复的数组之外,需要把前两个数定下来,然后转化为三数求和问题。

程序

class Solution {
public:
	/**
	* @param numbers: Give an array
	* @param target: An integer
	* @return: Find all unique quadruplets in the array which gives the sum of zero
	*/
	vector<vector<int>> fourSum(vector<int> &numbers, int target) {
		// write your code here
		//借助三个数之和的思想,此时时间复杂段为O(n^3),空间复杂度为O(1)
		vector<vector<int> > result;
		sort(numbers.begin(), numbers.end());
		for (int i = 0; i < numbers.size(); i++){//固定住第一个数
			if (i > 0 && numbers[i] == numbers[i - 1])
				continue;
			for (int j = i + 1; j < numbers.size(); j++){//固定住第二个数
				if (j > i + 1 && numbers[j] == numbers[j - 1])
					continue;
				for (int k = j + 1, h = numbers.size() - 1; k < h;){
					if (k > j + 1 && numbers[k] == numbers[k - 1]){
						k++;
						continue;
					}
					if (h < numbers.size() - 1 && numbers[h] == numbers[h + 1]){
						h--;
						continue;
					}
					int sum = numbers[i] + numbers[j] + numbers[k] + numbers[h];
					if (sum < target)
						k++;
					else if (sum > target)
						h--;
					else
					{
						vector<int> v(4, numbers[i]);
						v[1] = numbers[j];
						v[2] = numbers[k++];
						v[3] = numbers[h--];
						result.push_back(v);
					}
				}
			}
		}
		return result;
	}
};


猜你喜欢

转载自blog.csdn.net/qq_18124075/article/details/80516079