1. Two Sum 兩數之和 來源:力扣(LeetCode)
題目:給定一個整數數組和一個目標值,找出數組中和為目標值的兩個數。你可以假設每個輸入只對應一種答案,且同樣的元素不能被重復利用。
示例: 給定 nums = [2, 7, 11, 15], target = 9 因為 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
這道題目比較容易想到的是通過暴力搜索求解,但暴力搜索的時間復雜度為 O(n^2)。如果使用哈希的思想,利用Python中list的查詢方式,我們可以將復雜度降低到 O(n)。有兩個值得注意的地方:
(1). 同樣的元素不能重復使用(也就是不能自己加自己) (2). 給定的數組中可能有相同的元素(比如 [3, 3, 4, 4, 5])
java代碼:
public class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer, Integer> m = new HashMap<Integer, Integer>();
int[] res = new int[2];
for (int i = 0; i < nums.length; ++i) {
if (m.containsKey(target - nums[i])) {
res[0] = i;
res[1] = m.get(target - nums[i]);
break;
}
m.put(nums[i], i);
}
return res;
}
}
HashMap是Java中集合的一部分。它提供了Java的Map接口的基本實現。它將數據存儲在(Key,Value)對中。
- void clear():用於從Map中刪除所有映射。
- boolean containsKey(Object key):用於返回True如果對於指定的鍵,則映射存在於映射中。
- boolean containsValue(Object value):用於在一個或多個鍵映射到指定值時返回true。
- boolean isEmpty():用於檢查映射是否為空。如果地圖為空,則返回true。
- set entrySet():用於返回哈希映射的set視圖。
- Object get(Object key):用於檢索或獲取特定鍵映射的值。
- Set keySet():用於返回鍵的設置視圖。
- int size():用於返回Map的大小。
- Object put(Object key,Object value):用於將鍵值對的特定映射插入到映射中。
- putAll(Map M):用於將所有元素從一個Map復制到另一個Map。
- Object remove(Object key):用於刪除Map中任何特定鍵的值。
python代碼:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
numsMap = {}
for index,num in enumerate(nums):
tmpDiff = target-num
if tmpDiff in numsMap.keys():
return [numsMap[tmpDiff],index]
numsMap[num] = index
return None