【icyle】Leetcode-cn：1.两数之和

题目

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 的那 两个 整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。
你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

解答

思路

遍历数组，对于当前的nums[i]，找其他的target-nums[i]。

• 若找到，直接返回下标。

• 若找不到，用一对一hash将当前nums[i]和下标存入map中。

需要的头文件

map、vector、algorithm

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <map>
#include <algorithm>

using namespace std;

class Solution
{
public:
    vector<int> twoSum(vector<int> &nums, int target)
    {
        //定义存放key-num的hash地图
        map<int, int> key_num_map;

        //定义返回数组(现在是一个空数组)
        vector<int> reloc;

        //遍历一次数组，如果数组中存在target-nums[i]的数，则直接返回
        for (int i = 0; i < nums.size(); i++)
        {
            int tg = target - nums[i];
            vector<int>::iterator result = find(nums.begin(), nums.end(), tg);
            //返回找到元素的位置
            int loc = result - nums.begin();
            //当目前找到的数在数组中且不是同一个数时（同一个数不能用两遍）
            if (result == nums.end() || loc == i)
            {
                continue;
            }
            else if (loc != i)
            {
                reloc.push_back(i);
                reloc.push_back(loc);
                return reloc;
            }
        }

        //不存在这样的数对，返回空
        return {};
    }
};

时间复杂度和空间复杂度

• 时间复杂度：1个for循环，所以为 O(N) 。寻找target-nums[i]的复杂度由暴力算法的 O(N) 降为 O(1) 。
• 空间复杂度：定义了一个hash表，大小不超过数组长度。所以为 O(N) 。

反思与总结

1. 当查找不到结果时，记住需要返回合适的结果，如此处的空数组或报错语句。
2. vector返回数组元素下标的方法：迭代器相减