两数之和变体 (阿里面试)

题目:

给定一个整数数组和一个目标值,找出数组中和最接近目标值的两个数。

思路:

还是类似寻找两数之和的思路,用前后两个指针,不同的是需要新增加一个变量保存和与target之间的最小差值dis,以及最小差值对应的两个数。这个差值先初始化一下,每次先计算一下当前的两个数与target的距离,如果当前差值小于最小差值,更新最小差值,并且用当前两个数更新result,如果有当前两个数的和正好等于target的情况,直接输出result即可,这一定是最接近的,如果不等,就用寻找相等的思路移动前后两个指针,使得两个数之和往更接近target的方向走。如果没有正好相等的,就输出最后保存的最接近的两个数。

vector<int> twoSumClose(vector<int>& nums, int target)
{
	vector<int> result;
	if(nums.empty()||nums.size()<2)
		return result;
	sort(nums.begin(),nums.end())
	int i=0;
	int j=nums.size()-1;
	int dis=abs(nums[0]+nums[1]-target);
	result.push_back(nums[0]);
	result.push_back(nums[1]);
	while(i<j)
	{
		int temp=abs(nums[i]+nums[j]-target);
		if(temp<dis)
		{
			dis=temp;
			result[0]=nums[i];
			result[1]=nums[j];
		}
		if(nums[i]+nums[j]<target)
			i++;
		else if(nums[i]+nums[j]>target)
			j--;
		else
			return result;
	}
	return result;
}

类似的,还有最接近的三数之和,只需要增加一个循环。这里只需要找到一组解输出即可,所以不需要去重。

vector<int> threeSumClose(vector<int>& nums, int target)
{
	vector<int> result;
	if(nums.empty()||nums.size()<3)
		return result;
	sort(nums.begin(),nums.end());
	int dis=abs(nums[0]+nums[1]+nums[2]-target);
	result.push_back(nums[0]);
	result.push_back(nums[1]);
	result.push_back(nums[2]);
	for(int k=0;k<nums.size()-2;k++)
	{
		int i=k+1;
		int j=nums.size()-1;
		while(i<j)
		{
			int temp=abs(nums[i]+nums[j]+nums[k]-target);
			if(temp<dis)
			{
				dis=temp;
				result[0]=nums[k];
				result[1]=nums[i];
				result[2]=nums[j];
			}
			if(nums[i]+nums[j]<target-nums[k])
				i++;
			else if(nums[i]+nums[j]>target-nums[k])
				j--;
			else
				return result;
		}
	}
	return result;
}

猜你喜欢

转载自blog.csdn.net/u012991043/article/details/81286049