day_8——LeetCode1：兩數之和

Day_8 —— LeetCode練習題：兩數之和

1. 兩數之和

給定一個整數數組 nums 和一個目標值 target，請你在該數組中找出和爲目標值的那兩個整數，並返回他們的數組下標。

你可以假設每種輸入只會對應一個答案。但是，數組中同一個元素不能使用兩遍。

示例：

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

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

1. 首先，我們先遍歷數組nums，i爲當前元素下標，我們需要將每一個遍歷的值放入map中做爲key，元素下標作爲value；
2. 同時，對每一個元素都判斷map中是否存在target-nums[i]的key值。這裏的話，9-7=2，2在map中已存在；
3. 所以2和7所在的key值對應的value，也[0,1]，就是我們要的兩個數組下標。

C++實現：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) 
    {
        //建立一對一的hash
        map<int,int> a_map;
        //用來存儲找到的結果下標，初始化一個大小爲2，值爲0的容器b
        vector<int> b(2,0);
        int size=nums.size();
        for(int i=0;i<size;++i)
        {
            if(a_map.count(target-nums[i])>0)
            {
                b[0]=a_map[target-nums[i]];
                b[1]=i;
                break;
            }
            a_map[nums[i]]=i;//反過來放入map中，用來獲取結果下標
        }
        return b;
    };
};

python實現：

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        length = len(nums)
        dict_nums = {nums[i]:i for i in range(length)}
        for i in range(length):
            a = nums[i] # key值
            b = target - a 
            if b in dict_nums and i != dict_nums[b]:
                return [i, dict_nums[b]]