1. 兩數之和
- 2022 年 2 月 9 日
- 筆記
- LeetCode每日一題
給定一個整數數組 nums
和一個整數目標值 target
,請你在該數組中找出 和為目標值 target
的那 兩個 整數,並返回它們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,數組中同一個元素在答案里不能重複出現。
你可以按任意順序返回答案。
示例 1:
輸入:nums = [2,7,11,15], target = 9 輸出:[0,1] 解釋:因為 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
輸入:nums = [3,2,4], target = 6 輸出:[1,2]
示例 3:
輸入:nums = [3,3], target = 6 輸出:[0,1]
提示:
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
- 只會存在一個有效答案
暴力猜解:(最傳統的兩層for循環)
class Solution { public int[] twoSum(int[] nums, int target) { int n = nums.length; for (int i = 0; i < n; ++i) { for (int j = i + 1; j < n; ++j) { if (nums[i] + nums[j] == target) { return new int[]{i, j}; } } } return new int[0]; } }
進階:你可以想出一個時間複雜度小於 O(n2)
的算法嗎?
絕妙解法:
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> map = new HashMap<>();
for(int i = 0; i < nums.length; i++){
if(map.containsKey(nums[i])){ //for的數剛好是之前put期待的數
return new int[]{map.get(nums[i]), i};
}
map.put(target - nums[i], i); //暫時不滿足,放入期待的數
}
return null;
}
}