给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路:涉及到元素查找问题,而且考虑到要返回下标,所以采用unordered_map结构,可以同时存储key与value;
#include<iostream>
#include<vector>
#include<string.h>
#include<unordered_map>
using namespace std;
vector<int> sum_two(vector<int>& nums1, int target)
{
unordered_map<int,int>map;
for(int i=0;i<nums1.size();i++)
{
if(map.find(target-nums1[i])!=map.end())
{
return{map.find(target-nums1[i])->second,i};
}
map.insert(pair<int,int>(nums1[i],i));
}
return {};
}
void print(vector<int>&C)
{
for(int i=0;i<C.size();i++)
{
cout<<C[i]<<' ';
cout<<endl;
}
}
int main()
{
cout<<"输入A容器的大小:"<<endl;
int A_len;
cin>>A_len;
vector<int>A;
cout<<"输入A元素"<<endl;
for (int i=0;i<A_len;i++)
{
int a;
cin>>a;
A.push_back(a);
}
cout<<"输入元素target:"<<endl;
int B;
cin>>B;
vector<int>C=sum_two(A,B);
if (C.empty())
{
cout<<"没有符合要求的元素"<<endl;
return 0;
}
cout<<"元素下标为:"<<endl;
print(C);
return 0;
}
运行结果:
输入A容器的大小:
4
输入A元素
1
2
3
4
输入元素target:
9
没有符合要求的元素
输入A容器的大小:
5
输入A元素
1
2
2
3
1
输入元素target:
5
元素下标为:
1
3