1. 程式人生 > 實用技巧 >【leetcode每日兩題】-Day1-簡單題

【leetcode每日兩題】-Day1-簡單題

1. 兩數之和

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

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

示例:

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

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

class Solution:
    #雜湊
    def twoSum(self, nums: List[int], target: int) -> List[int]: # nums = [2, 7, 11, 15] target = 9
hashmap = {} for i in range(len(nums)): # i= 0,1,2,3 if (target - nums[i]) in hashmap: # target = 9 target -nums[i] = 7 , 2 , -2 , -6 return [nums.index(target - nums[i]),i] # 錯誤:hashmap.get(i) = nums[i] # {0:2,1:7,2:11,3:15} hashmap[nums[i]] = i

做題思路:

  本題要求我們根據target返回nums中符合條件的索引並放在一個list中,考慮到這裡,我們可以考慮雜湊表,因為雜湊表這種key-value形式可以減小資料查詢的時間複雜度,將時間複雜度從兩層迴圈的o(n2)降低為o(n)。程式碼中

hashmap[nums[i]] = i就是給雜湊表中的kay-value賦值。需要注意的是python語言不想java等其他語言需要宣告資料型別,所以在python做題當中經常用dict類比hashmap,用list類比queue和stack等等

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/two-sum

著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。