每天一道算法题（二）：给定数组Arr和一个整数aim,请返回哪两个位置的数可以加出aim来。

最新推荐文章于 2021-09-17 01:00:52 发布

Dawn丶

最新推荐文章于 2021-09-17 01:00:52 发布

阅读量621

点赞数

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/weixin_40222745/article/details/79235567

版权

算法专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种解决两数之和问题的有效算法。通过排序和双指针技巧，快速定位到目标数值对应的两个元素。文章提供了完整的C++实现代码，并采用辅助数组保持原始索引，确保了解决方案的正确性和效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定数组Arr和一个整数aim,请返回哪两个位置的数可以加出aim来

例如：arr={2,7,11,15},target = 9 返回{0，1}，因为arr[0] + arr[1] = 2 + 7 = 9,可以假设每个数组里只有一组答案。

大概流程是这样的：

首先，定义一个数组。记录数组的位置。将数组进行排序，在排序的过程中，如果值交换，则将保存位置的数组也随之交换

我们可以定义两个变量。一个指向数组的起始位置，一个指向数组末尾位置。

如果起始位置和末尾位置相加值大于aim,则末尾位置--。

如果起始位置和末尾位置相加值小aim,则起始位置++。

如果相等，返回数组内的原来的位置。

详细代码如下：

#include <iostream>
#include <vector>
using namespace std;

//如果交换，将辅助数组和原数组都要进行交换
void swap(vector<int> &nums, vector<int> &help,int m, int n)
{
	int tmp = nums[m];
	nums[m] = nums[n];
	nums[n] = tmp;
	
	tmp = help[m];
	help[m] = help[n];
	help[n] = tmp;
}


//构建最大堆
void maxHeap(vector<int> &nums, int start, int end, vector<int> &help)
{
	int fNode = start;
	int sNode = 2*start+1;

	while (sNode <= end)
	{
		if (sNode+1 <= end && nums[sNode] < nums[sNode+1])
			sNode++;
		if (nums[fNode] > nums[sNode]) 
			return;
		else
		{
			swap(nums,help,sNode,fNode);
			fNode = sNode;
			sNode = fNode*2+1;
		}
	}

}


//堆排序
void heapSort(vector<int> &nums, vector<int> &help)
{
	int len = nums.size();
	for(int i = len/2-1; i >= 0; i--)
	{
		maxHeap(nums, i, len-1, help);
	}
	for(int j = len-1; j >= 0; j--)
	{
		swap(nums, help, j ,0);
		maxHeap(nums,0,j-1, help);
	}
}


void twoNum(vector<int> &nums, vector<int> &help, vector<int> &res, int aim)
{
	heapSort(nums,help);
	int left = 0;
	int right = nums.size()-1;
	int sum = 0;
	while (left != right)//排序完以后进行查找
	{
		sum = nums[left] + nums[right];
		if (sum > aim)
			right--;
		if (sum < aim) 
			left++;
		if (sum == aim) 
		{
			res.push_back(left);
			res.push_back(right);
			return;
		}
	}
	res.push_back(-1);
	res.push_back(-1);
}


int main(int argc, char* argv[])
{
	vector<int> nums;
	nums.push_back(2);
	nums.push_back(11);
	nums.push_back(7);
	nums.push_back(15);
	
	vector<int> help;
	
	vector<int> res;

	for(int i = 0; i < 4; i++)
	{
		help.push_back(i);
	}

	twoNum(nums, help, res,26);

	for(i =0; i < 2; i++)
	{
		cout << res[i] << "\t" ;
	}
	cout << endl;

	cin.get();
	return 0;
}