LeetCode刷題--兩數之和（簡單）

題目描述

給定一個整數數組 nums 和一個目標值 target，請你在該數組中找出和爲目標值的那 兩個 整數，並返回他們的數組下標。

你能夠假設每種輸入只會對應一個答案。可是，你不能重複利用這個數組中一樣的元素。

示例:

給定 nums = [2, 7, 11, 15], target = 9

由於 nums[0] + nums[1] = 2 + 7 = 9
因此返回 [0, 1]

 

方法一：暴力法

思路

 暴力法很簡單，遍歷每一個元素 x，並查找是否存在一個值與 target − x 相等的目標元素。

代碼

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（n2），對於每一個元素，咱們試圖經過遍歷數組的其他部分來尋找它所對應的目標元素，這將耗費O（n）的時間。所以時間複雜度爲O（n2）。·

• 空間複雜度：O（1）。

 

方法二：一遍哈希表（優質解法）

思路

利用HashMap 減小查詢時間

• 在暴力法中，內層循環查找差值很浪費時間，那麼如何減小查詢時間呢？利用HashMap就能夠減小查詢時間。

• 使用一層循環，每遍歷到一個元素就計算該元素與target之間的差值，而後到HashMap中尋找該差值，若是找到，則返回兩個元素在數組nums的下標，若是沒有找到，則將當前元素存入HashMap中（Key:nums [i] ，Value : i ）。

代碼

class Solution { public int[] twoSum(int[] nums, int target) { HashMap<Integer,Integer> map = new HashMap<>(); int[] res = new int[2]; for (int i = 0; i < nums.length; i++) { int dif = target - nums[i]; if (map.get(dif) != null) { res[0] = map.get(dif); res[1] = i; return res; } map.put(nums[i],i); } return res; } }

 

複雜度分析

• 時間複雜度：O（n），咱們只遍歷了包含有n個元素的列表一次。在表中進行的每次查找只花費O（1）的時間。

• 空間複雜度：O（n），所需的額外空間取決於哈希表中存儲的元素數量，該表最多須要存儲n個元素。

代碼優化

public class Solution { public int[] twoSum(int[] nums, int target) { Map<Integer, Integer> map = new HashMap(); for (int i = 0; i < nums.length; ++i) { if (map.containsKey(target- nums[i])) { return new int[]{map.get(target- nums[i]), i}; } map.put(nums[i], i); } return int[]{-1, -1}; } }

 

思考

看到這個題，第一個想到的就是暴力法，確實作出來了，發現時間複雜度和空間複雜度都挺高的，hashMap的時間複雜度遠遠低於暴力法，算是用空間換時間的一種方法了。

代碼優化之後儘量的去作，話說討論區好多大佬啊。