關於兩數之和的算法


介紹一下簡單的算法:

給定一個整數數組和一個目標值,找出數組中和為目標值的兩個數。

你可以假設每個輸入只對應一種答案,且同樣的元素不能被重復利用。

示例:

對於這道題,首先想到的就是暴力方法,即使用兩個for循環,遍歷兩次數組,看有沒有和是目標值的。顯然這樣時間復雜度太大,O(n*n)

因此需要改良的算法,用到哈希查找的方法。

建立哈希表,從左向右掃描一遍,將整數與索引存放到map中。掃描一遍,對其中的每一個整數K,搜索 target-K 在map中是否存在即可。若存在,則輸出 K 與 target-K 的下標即可。此算法的時間復雜度為O(n)

語言:java

public int[] twoSum(int[] nums, int target) {
    Map<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        int complement = target - nums[i];
        if (map.containsKey(complement)) {
            return new int[] { map.get(complement), i };
        }
        map.put(nums[i], i);
    }
    throw new IllegalArgumentException("No two sum solution");
}

注意:轉化思想,與其一個個比較數組中的兩數之和是否為目標值不如確定一個加數,使用目標值減去其得到另一個加數,看是否在數組中存在。

 
       


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM