從今天開始,寫一下我在刷 LeetCode 時的心得體會,包括自己的思路和別人的優秀思路,歡迎各種監督啊! 2016/10/9
編程語言是 Java,代碼托管在我的 GitHub 上,包括測試用例。歡迎各種批評指正!
<br />
題目 —— Two Sum
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution.
Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
<br >
解答
題目大意
給出一個整型數組,給出一個特定的數 target,返回和等于 target 的兩數下標,假設每次輸入只有一組解。-
解題思路
- 首先,我能想到最簡單的思路就是兩個 for 循環,遍歷整個數組,n 平方的時間復雜度,這個思路就不再寫了,很容易。
-
看了一下標簽,發現是
Array
和Hash Table
,結合討論區的朋友們解答,理解了如下思路:- 創建一個 int 數組 result 存放需要返回的下標,因為只需要返回兩個,所以數組大小為 2。
- 創建一個 HashMap<Integer, Integer> 類型的變量 map,key 存放數值,value 存放下標。
- 循環遍歷數組 nums,判斷在 map 中是否存在 target-nums[i]:若存在,直接將兩個下標存入 result 數組,并返回;若不存在,將 nums[i] 的值和下標放入 map,進入下一次循環。
代碼實現
public class Solution {
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(target - nums[i])) {
result[1] = i;
result[0] = map.get(target - nums[i]);
return result;
}
map.put(nums[i], i);
}
return result;
}
}
-
小結
用 HashMap 解決問題的好處是,只需要遍歷一次數組,時間復雜度為 O(n)。但難點在于,不容易想到這種方法。我們考慮從每次輸入只有一組解
這句話入手,所以數組中肯定沒有重復的值,從而使問題迎刃而解。
另外有個小細節,如果想順序輸出下標的話,需要把 i 賦值給 result[1],因為我們在向 map 中添加元素的時候,是順序添加的,這個可以自己輸入體會一下。