每日一個小算法之兩數之和
給定一個整數數組 nums 和一個目標值 target,請你在該數組中找出和爲目標值的那 兩個 整數,並返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個數組中同樣的元素。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因爲 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1].
這個題目是不是很簡單啊。
本文出自凱哥Java(kaigejava)
首先,我們想到的第一種方案就是for循環。嵌套循環。如下:
JAVA代碼:
public static void main(String[] args) { Integer [] nums = new Integer []{1,2,0,5,7,9,3}; Integer target = 5; Integer[] returnNums = twoSum(nums,target); System.out.println(JSON.toJSONString(returnNums)); } public static Integer[] twoSum(Integer[] nums, Integer target) { Integer[] returnNums = new Integer[2]; for (int i = 0; i < nums.length; i++) { for (int j = i+1; j < nums.length; j++) { int num1 = nums[i]; int num2 = nums[j]; if ((num1 + num2) == target) { returnNums[0] = i; returnNums[1] = j; return returnNums; } } } return returnNums; }
運行結果:
執行結果是出來了。我們看看執行時間:
是不是很low的一種方法。
那麼接下來,我們看看第二中算法:
第二種算法和巧妙。Java代碼如下:
private Integer[] twoSum2(Integer[] nums, Integer target) { Integer[] indexs = new Integer[2]; // 建立k-v ,一一對應的哈希表 HashMap<Integer,Integer> hash = new HashMap<Integer,Integer>(); for(int i = 0; i < nums.length; i++){ if(hash.containsKey(nums[i])){ indexs[0] = i; indexs[1] = hash.get(nums[i]); log.info(JSON.toJSONString(hash)); return indexs; } // 將數據存入 key爲補數 ,value爲下標 hash.put(target-nums[i],i); log.info("==>:{},i=:{},target-nums[i]:{}",nums[i],i,target-nums[i]); } return indexs; }
其中巧妙的地方:
再來看看運行結果:
看看兩個算法對比:
再速度上是不是快了很多。
歡迎大家留下更好的算法。一起學習,一起成長