給定一個整數數組 nums 和一個目標值 target,請你在該數組中找出和爲目標值的那 兩個 整數,並返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,數組中同一個元素不能使用兩遍。
示例:
給定 nums = [2, 7, 11, 15], target = 9
返回 [0, 1]
方法一:最簡單的暴力兩次for循環
class Solution {
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^2)
對於每個元素,我們試圖通過遍歷數組的其餘部分來尋找它所對應的目標元素,這將耗費 O(n)的時間。因此時間複雜度爲 O(n^2)。
空間複雜度:O(1)
方法二:使用HashMap遍歷
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int poor = target - nums[i];
if (map.containsKey(poor)) {
return new int[] { map.get(poor), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
}
複雜度分析:
時間複雜度:O(n)
我們只遍歷了包含有 nn個元素的列表一次。在表中進行的每次查找只花費 O(1)的時間。
空間複雜度:O(n)
所需的額外空間取決於哈希表中存儲的元素數量,該表最多需要存儲 nn個元素。