算法在編程領域的重要性不言而喻,而且也是好多大廠面試經常要考覈的重點。
兩數之和
題目:給定一個整數數組 nums 和一個目標值 target,請你在該數組中找出和爲目標值的那兩個整數,並返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個數組中同樣的元素。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因爲 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
算法思路:
- 暴力法:使用兩層循環,外層循環計算當前元素與 target 的差值,內層循環尋找該差值,若找到該差值,則返回兩個元素的下標;時間複雜度:O(n^2)
- 利用HashMap減少查詢時間:HashMap 可以根據下標快速查找數據,減少了查詢時間,結合這個特性使用一層循環即可以實現:每遍歷一個元素就計算該元素與 target 之間的差值,然後到 HashMap 中查找該差值,如果找到則返回兩個元素的下標,如果沒有找到,則將元素存入 HashMap 中(Key:nums[i],Value:i);時間複雜度:O(n)。
算法代碼:根據算法思路2,寫出的算法具體代碼如下:
/**
* 兩數之和
* @param nums
* @param target
* @return
*/
public int[] twoSum(int[] nums, int target) {
HashMap<Integer, Integer> map = new HashMap<>();
int[] res = new int[2];
for (int i = 0; i < nums.length; i ++) {
int value = target - nums[i];
if (map.get(value) != null) {
res[0] = map.get(value);
res[1] = i;
}
map.put(nums[i], i); // 構建 HashMap
}
return res;
}