介紹一下簡單的算法:
給定一個整數數組和一個目標值,找出數組中和為目標值的兩個數。
你可以假設每個輸入只對應一種答案,且同樣的元素不能被重復利用。
示例:
對於這道題,首先想到的就是暴力方法,即使用兩個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");
}
注意:轉化思想,與其一個個比較數組中的兩數之和是否為目標值不如確定一個加數,使用目標值減去其得到另一個加數,看是否在數組中存在。