欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Leetcode1—两数之和C++实现

程序员文章站 2022-07-14 17:28:56
...

Leetcode1—两数之和C++实现

题目:给定一个数组,和一个目标值target,遍历数组中的元素,找到两个元素之和恰好等于target,返回两个元素的下标。
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

解题思路:

设置两个空vector,一个将原数组中的元素全部拷贝过来,进行遍历查找,其中采用双指针的方法:指针i指向首元素,指针j指向尾元素,找到两个元素和等于target之后,将这两个元素存到空的vector中返回。

具体实现:

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> Twosum(vector<int>& arr, int target)
{
	vector<int>ans;
	vector<int>temp;
	temp = arr;
	int n = temp.size();
	sort(temp.begin(),temp.end());
	int i = 0; int j = n - 1;
	while (i < j)
	{
		int sum = temp[i] + temp[j];
		if (sum > target)
			j--;
		else if (sum < target)
			i++;
		else break;
	}
	if (i < j)
	{
		for (int k = 0; k < n; k++)
		{
			if (i<n &&arr[k] == temp[i])//这里是个讲究,顺序不能颠倒
			{
				ans.push_back(k);
				i = n;
			}
			else if (j<n &&arr[k] == temp[j] )
			{
				ans.push_back(k);
				j = n;
			}
			if (i == n && j == n)
				return ans;
		}
	}
	return ans;
}
int main()
{
	vector<int>arr = { 1,9,7,8,5,4,6,3,2 };
	vector<int>temp=Twosum(arr, 10);
	for (vector<int>::iterator it=temp.begin();it!=temp.end();it++)
	{
		cout << *it << " ";
	}
	cout << endl;
}

输出如下:

Leetcode1—两数之和C++实现