354-Leetcode 四数之和

class Solution 
{
public:
	vector<vector<int>> fourSum(vector<int>& nums, int target)
	{
		vector<vector<int>> vec;
		if (nums.size() < 4) 
		{
			return vec;
		}
		sort(nums.begin(), nums.end());
		int length = nums.size();
		for (int i = 0; i < length - 3; i++)
		{
			if (i > 0 && nums[i] == nums[i - 1]) 
			{
				continue;
			}
			if ((long)nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) 
			{
				break;
			}
			if ((long)nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target) 
			{
				continue;
			}
			for (int j = i + 1; j < length - 2; j++) 
			{
				if (j > i + 1 && nums[j] == nums[j - 1]) 
				{
					continue;
				}
				if ((long)nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) 
				{
					break;
				}
				if ((long)nums[i] + nums[j] + nums[length - 2] + nums[length - 1] < target) 
				{
					continue;
				}
				int left = j + 1, right = length - 1;
				while (left < right) 
				{
					int sum = nums[i] + nums[j] + nums[left] + nums[right];
					if (sum == target) 
					{
						vec.push_back({ nums[i], nums[j], nums[left], nums[right] });
						while (left < right && nums[left] == nums[left + 1]) 
						{
							left++;
						}
						left++;
						while (left < right && nums[right] == nums[right - 1]) 
						{
							right--;
						}
						right--;
					}
					else if (sum < target) 
					{
						left++;
					}
					else 
					{
						right--;
					}
				}
			}
		}
		return vec;
	}
};
int main()
{
	Solution A;
	vector<int> vec1{ 1,0,-1,0,-2,2 };
	vector<vector<int>> vec2 = A.fourSum(std::ref(vec1), 0);
	return 0;
}

时间复杂度：O(n^ 3)，其中 n 是数组的长度。排序的时间复杂度是 O(nlogn)，枚举四元组的时间复杂度是 O(n^ 3)，因此总时间复杂度为 O(n^ 3+nlogn)=O(n^ 3)

空间复杂度：O(logn)，其中 n 是数组的长度。空间复杂度主要取决于排序额外使用的空间。此外排序修改了输入数组 nums，实际情况中不一定允许，因此也可以看成使用了一个额外的数组存储了数组 nums 的副本并排序，空间复杂度为 O(n)