18四数之和(双指针法)

1、题目描述

给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:

答案中不可以包含重复的四元组。

2、示例

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

3、题解

采用双指针法:灵感来自两数之和,先对nums数组排序,然后固定nums[t]和nums[k],对k后面两端设定两个指针i=k+1和j=len-1,然后判断nums[t]+nums[k]+nums[i]+nums[j]四数之和与target关系

  • 如果四数之和>target, j--指针左移
  • 如果四数之和<target, i++指针右移
  • 如果四数之和=target, 保存四个数到res, i++ j--

这个问题还需要防止出现重复的四元组

  • 防止出现重复的第一个操作:对于nums[i]+nums[j]+nums[k]+nums[t]=target,nums[t]相连元素可能相同跳过
  • 防止出现重复的第二个操作:对于nums[i]+nums[j]+nums[k]+nums[t]=target,nums[k]相连元素可能相同跳过
  • 防止出现重复的第三个操作:对于nums[i]+nums[j]+nums[k]+nums[t]=target,nums[i]与nums[j]可能与nums[i+1]与nums[j-1]相连元素可能相同跳过
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution1 {
public:
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
		int i, j, k, t;
		vector<vector<int>> res;
		int len = nums.size();
		if (len < 4)  //当输入的元素小于4,返回""
			return res;
		sort(nums.begin(), nums.end());  //对nums数组元素排序
		//采用双指针法:灵感来自两数之和,先对nums数组排序,然后固定nums[t]和nums[k],对k后面两端设定两个指针i=k+1和j=len-1,然后判断nums[t]+nums[k]+nums[i]+nums[j]四数之和与target关系
		//如果四数之和>target, j--指针左移
		//如果四数之和<target, i++指针右移
		//如果四数之和=target, 保存四个数到res, i++ j--
		for (t = 0; t < len - 3; t++)
		{
			//防止出现重复的第一个操作:对于nums[i]+nums[j]+nums[k]+nums[t]=target,nums[t]相连元素可能相同跳过
			if (t > 0 && nums[t] == nums[t - 1])
				continue;
			for (k = t + 1; k < len - 2; k++)
			{
				//防止出现重复的第二个操作:对于nums[i]+nums[j]+nums[k]+nums[t]=target,nums[k]相连元素可能相同跳过
				if (k > t + 1 && nums[k] == nums[k - 1])
					continue;
				i = k + 1;
				j = len - 1;
				while (i < j)
				{
					if (nums[i] + nums[j] + nums[k] + nums[t] > target)
						j--;
					else if (nums[i] + nums[j] + nums[k] + nums[t] < target)
						i++;
					else
					{
						res.push_back({ nums[t],nums[k],nums[i],nums[j] });
						//防止出现重复的第三个操作:对于nums[i]+nums[j]+nums[k]+nums[t]=target,nums[i]与nums[j]可能与nums[i+1]与nums[j-1]相连元素可能相同跳过
						while (i < j && nums[i] == nums[i + 1])
							i++;
						while (i < j && nums[j] == nums[j - 1])
							j--;
						i++;
						j--;
					}
				}
			}
		}
		return res;
	}
};
//解法二主要是用到STL库中erase和unique函数去重复,代码简单一点,不用像解法二那样去重复的四元组
class Solution2 {
public:
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
		int i, j, k, t;
		vector<vector<int>> res;
		int len = nums.size();
		if (len < 4)
			return res;
		sort(nums.begin(), nums.end());
		for (t = 0; t < len - 3; t++)
		{
			for (k = t + 1; k < len - 2; k++)
			{
				i = k + 1;
				j = len - 1;
				while (i < j)
				{
					if (nums[i] + nums[j] + nums[k] + nums[t] > target)
						j--;
					else if (nums[i] + nums[j] + nums[k] + nums[t] < target)
						i++;
					else
					{
						res.push_back({ nums[t],nums[k],nums[i],nums[j] });
						i++;
						j--;
					}
				}
			}
		}
		sort(res.begin(), res.end());
		res.erase(unique(res.begin(), res.end()), res.end());
		return res;
	}
};
int main()
{
	Solution1 solute;
	vector<vector<int>> res;
	vector<int> nums = { -3,-2,-1,0,0,1,2,3 };
	int target = 0;
	res = solute.fourSum(nums, target);
	for (int i = 0; i < res.size(); i++)
	{
		for (int j = 0; j < 4; j++)
			cout << res[i][j] << " ";
		cout << endl;
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值