GEEK编程练习— —两数求和

Posted 2020-07-13 Sin_Geek

题目

输入一个特定整数值和一组整数，要求从这组整数中找到两个数，使这两数之和等于特定值。按照从前往后的顺序，输出所有满足条件的两个数的位置。具体格式如下：

输入

9
1 2 4 5 7 9 11 

输出

2 5
3 4

分析

1)暴力解法，复杂度O（n^2），不考虑
2)hash。用哈希表存储每个数对应下标，复杂度O(n)
3)先排序，然后左右夹逼，排序O(nlogn)，夹逼O(n)，最终O(nlogn)。但是需要的是位置，而不是数字本身，此方法不适用。

代码

#include <iostream>
#include <vector>
#include <unordered_map>

using namespace std;

int main()
{
    vector<int> nums;
    int target;
    int tmp;
    cin >> target;
    while (cin >> tmp)
        nums.push_back(tmp);

    unordered_map<int, int> mapping;
    vector<int> result;
    for (auto i = 0; i < nums.size(); ++i)
    {
        mapping[nums[i]] = i;
    }

    for (auto i = 0; i < nums.size(); ++i)
    {
        const int gap = target - nums[i];
        if (mapping.find(gap) != mapping.end() && mapping[gap] > i)
        {
            result.push_back(i + 1);
            result.push_back(mapping[gap] + 1);
        }
    }

    for (auto i = 0; i < result.size(); i += 2)
    {
        cout << result[i] << ends << result[i + 1] << endl;
    }
    return 0;
}