題目:
給定一個整數數組 nums
和一個目標值 target
,請你在該數組中找出和爲目標值的那 兩個 整數,並返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個數組中同樣的元素。
示例:
給定 nums = [2, 7, 11, 15], target = 9 因爲 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
Java解法:
tip1:
暴力法
暴力法很簡單,遍歷每個元素 x,並查找是否存在一個值與 target - x 相等的目標元素。
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[j] == target - nums[i]) {
return new int[] { i, j };
}
}
}
throw new IllegalArgumentException("No two sum solution");
}
因爲是雙層循環,所以時間複雜度爲O(n²),而全局分配的空間沒有發生改變,所以空間複雜度爲O(1)。
tip2:
哈希算法
插入的表的同時,檢查是否有符合條件的值,如果有就返回下標。
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (map.containsKey(complement)) {
return new int[] { map.get(complement), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
-
時間複雜度:O(n)O(n), 我們只遍歷了包含有 nn 個元素的列表一次。在表中進行的每次查找只花費 O(1)O(1) 的時間。
-
空間複雜度:O(n)O(n), 所需的額外空間取決於哈希表中存儲的元素數量,該表最多需要存儲 nn 個元素。
PS:上面代碼塊是直接執行並返回的所不支持多條符合條件的數值,如下圖:
Python解法:
for i in range(len(nums)-1):
for j in range(i+1,len(nums)):
if nums[i]+nums[j] == target:
return [i,j]