【Leetcode】1.兩數之和

題目描述

鏈接：https://leetcode-cn.com/problems/two-sum

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

你可以假設每種輸入只會對應一個答案。但是，你不能重複利用這個數組中同樣的元素。

示例:

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

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

解題思路

題目很簡單，重點是怎麼提高效率

1. 暴力法：兩層循環解決，但是效率太低，時間複雜度爲O(n)

2. 單層循環+一個字典數據結構：循環的時候把相應的值存入到字典當中，如果用“目標值”-“當前值”在字典中，則符合條件

在開發中，除非是數據量特別大的情況下，都可以考慮用方法2，因爲相較於時間複雜度來說，空間複雜度可以放在第二位

 

代碼

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