從今天開始，寫一下我在刷 LeetCode 時的心得體會，包括自己的思路和別人的優(yōu)秀思路，歡迎各種監(jiān)督啊！ 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 >

解答

• 題目大意
  給出一個整型數(shù)組，給出一個特定的數(shù) target，返回和等于 target 的兩數(shù)下標(biāo)，假設(shè)每次輸入只有一組解。

• 解題思路

  • 首先，我能想到最簡單的思路就是兩個 for 循環(huán)，遍歷整個數(shù)組，n 平方的時間復(fù)雜度，這個思路就不再寫了，很容易。

• 看了一下標(biāo)簽，發(fā)現(xiàn)是 Array 和 Hash Table，結(jié)合討論區(qū)的朋友們解答，理解了如下思路：

  • 創(chuàng)建一個 int 數(shù)組 result 存放需要返回的下標(biāo)，因?yàn)橹恍枰祷貎蓚€，所以數(shù)組大小為 2。
  • 創(chuàng)建一個 HashMap<Integer, Integer> 類型的變量 map，key 存放數(shù)值，value 存放下標(biāo)。
  • 循環(huán)遍歷數(shù)組 nums，判斷在 map 中是否存在 target-nums[i]：若存在，直接將兩個下標(biāo)存入 result 數(shù)組，并返回；若不存在，將 nums[i] 的值和下標(biāo)放入 map，進(jìn)入下一次循環(huán)。

• 代碼實(shí)現(xiàn)

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;
    }
}

• 小結(jié)
  用 HashMap 解決問題的好處是，只需要遍歷一次數(shù)組，時間復(fù)雜度為 O(n)。但難點(diǎn)在于，不容易想到這種方法。我們考慮從 每次輸入只有一組解 這句話入手，所以數(shù)組中肯定沒有重復(fù)的值，從而使問題迎刃而解。
  另外有個小細(xì)節(jié)，如果想順序輸出下標(biāo)的話，需要把 i 賦值給 result[1]，因?yàn)槲覀冊谙?map 中添加元素的時候，是順序添加的，這個可以自己輸入體會一下。