ACM之兩數之和問題
阿新 • • 發佈:2018-04-21
ACM 兩數之和 Java 題目如下:
大概的意思是:給我們一個Int型的數組和一個目標值,找出數組中兩個值之和為目標值的元素位置;要求是每個輸入對應一個答案(即找到符合條件的元素直接返回結果就行,不用繼續往後找,難度降低)並且相同的元素值不能用兩次。
1.剛開始的思路:
偉大而又萬能的蠻力法,用兩個for循環,很容易求解,時間復雜度為O(N^2),代碼如下:
class Solution { public int[] twoSum(int[] nums, int target) { int[] res = new int[2]; for(int i = 0;i < nums.length;i++){ for(int j = i+1;j < nums.length;j++){ if((nums[i]+nums[j]) == target){ res[0] = i; res[1] = j; return res; } } } return res; } }
2.優化解法
蠻力法雖然是我這種菜鳥的最愛,可惜不是面試官要的答案,還是不能將就的。想辦法把時間復雜度將為O(N)就很爽了啊,也就是把上面蠻力法中的一個for循環給去掉,怎麽弄呢?蠻力法的思路是不斷地進行a+b去匹配目標值c,既然目標值知道,那我們當然可以選擇c-b去匹配a。用一個for循環將c-b的值存在一個數據結構中,然後查找這個數據結構中是否含有a就行。那麽問題來了,這個數據結構選什麽呢:數組?棧?隊列還是HashMap?一看前面三種選擇查找的時間復雜度還是O(N),而HashMap查找的時間復雜度為O(1),那肯定選HashMap,問題都搞定了,代碼實現出來就OK:
public class TwoSumBter { public static int[] twosum(int[] num,int target){ int[] res = new int[2]; int len = num.length; //判斷輸入數組的大小 if(num == null || num.length<2){ res[0] = 0; res[1] = 0; return res; } //建立HashMap HashMap<Integer,Integer> hm = new HashMap<Integer,Integer>(); for(int i=0;i<len;i++){ //判斷HashMap中是否含有目標數組中的數 if(hm.containsKey(num[i])){ res[0] = hm.get(num[i]); res[1] = i; return res; }else{ hm.put((target-num[i]), i); } } return new int[]{0,0}; } }
ACM之兩數之和問題