[LeetCode] 01 Two Sum 两数之和

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路:这道题要求不能重复,就一下排除了最好想到的暴力解法,遍历。随后考虑到每个元素只能利用一次,那么在读取每个元素时可以知道与之和成为目标值的对应值是多少(比如目标是12,第一个元素是5,那么对应的需要就是7),然后在后面的元素中判断中判断是否存在7. 由此想到用Map,利用count函数每次进行判断,将下标和值都保存起来。C++代码如下

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> m;
        for(int i=0;i<nums.size();++i)
        {
            if(m.count(target-nums[i]) )
                return {i,m[target-nums[i]]}; //注意这种返回的写法和顺序
            m[nums[i]]=i;
        }
        return {};
    }
};

猜你喜欢

转载自blog.csdn.net/annabelle1130/article/details/88528386