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++)