每日一個小算法之兩數之和

每日一個小算法之兩數之和

給定一個整數數組 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;
}

運行結果:

a31f9d367f105622468728959fd40c1b.png

執行結果是出來了。我們看看執行時間:

f4976f9f42f0a8dd4ecd8a43b8f16206.png

是不是很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;
}

其中巧妙的地方:

f26ec04957bb6939f2b89a7cade76b66.png

再來看看運行結果:

c17fb7dfbe2227becc7db56d498be764.png

看看兩個算法對比:

231c56bcd07165cfe6239de7adce2ded.png

再速度上是不是快了很多。

歡迎大家留下更好的算法。一起學習,一起成長


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章