歡迎來到力扣熱門100題系列文章!
在這個系列中,我將帶您逐一探索力扣平台上最受歡迎的100道題目。每篇文章都將包括問題的詳細説明、解題思路和代碼實現。希望通過這個系列,您能夠提升自己的算法和數據結構技能,為下一份工作做好準備。
1. 兩數之和
Question
https://leetcode.cn/problems/two-sum/
給定一個整數數組 nums 和一個整數目標值 target,請你在該數組中找出 和為目標值 target 的那 兩個 整數,並返回它們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,數組中同一個元素在答案裏不能重複出現。
你可以按任意順序返回答案。
示例 1:
<pre>輸入:nums = [2,7,11,15], target = 9
</pre>
輸出:
[0,1]
解釋:
因為 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
<pre>輸入:nums = [3,2,4], target = 6
</pre>
輸出:
[1,2]
示例 3:
<pre>輸入:nums = [3,3], target = 6
</pre>
輸出:
[0,1]
提示:
2 <= nums.length <= 104-109 <= nums[i] <= 109-109 <= target <= 109- 只會存在一個有效答案
進階:你可以想出一個時間複雜度小於 O(n2) 的算法嗎?
Solution
暴力法
可以使用嵌套循環查找滿足條件的答案。
class Solution {
public int[] twoSum(int[] nums, int target) {
for(int i = 0; i = nums.length; i++) {
for(int j = 1; j = nums.length; j++) {
if(nums[i] + nums[j] == target) {
return new int[]{i,j};
}
}
}
return null;
}
}
時間複雜度
O(n^2), n是數組nums的長度。
空間複雜度
O(1), 不需要使用額外空間。
使用Map
我們聲明一個Map,用來存儲數字和該數字對應下標。
我們遍歷數組,每次遍歷時用target減去nums[i]得到剩餘數字,然後查找剩餘數字是否在Map中。如果在Map,則返回剩餘數字的下標和當前下標;如果不在Map中,則將當前數字和當前下標寫入Map。
class Solution {
public int[] twoSum(int[] nums, int target) {
var map = new HashMap<Integer, Integer>();
for(var i = 0; i < nums.length; i++) {
var remain = target - nums[i];
if(map.containsKey(remain)) {
return new int[] {map.get(remain), i};
}
map.put(nums[i], i);
}
return null;
}
}
時間複雜度
O(n), n是數組nums的長度。
空間複雜度
O(n), n是數組nums的長度。