leetcode1. 两数之和

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

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

 

示例:

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

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

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        // 边角问题，我们要考虑边角问题如何处理
        vector<int> ret;
        if(nums.size()<=1)
            return ret;
        //新建一个map<key,value> 模式来存储number里的元素和index
        //这里的unordered_map 相当于hash_map
        unordered_map<int,int> myMap;
        for (int i = 0;i<nums.size();++i)
            myMap[nums[i]] = i;
        for (int i= 0;i<nums.size();i++){
            int rest_val = target- nums[i];
            if(myMap.find(rest_val)!=myMap.end()){
                int index = myMap[rest_val];
                if (index==i)
                    continue;
                if (index<i)
                {
                    ret.push_back(index);
                    ret.push_back(i);
                    return ret;
                }
                else{
                    ret.push_back(i);
                    ret.push_back(index);
                    return ret;
                }
            }
        }
        return ret;
    }
};

python3 版本

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dic = {}
        for i , num in enumerate(nums):
            if num in dic:
                return [dic[num], i]
            else:
                dic[target-num] = i