博客

给定一个整数数组 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