1. 程式人生 > >LeetCode 兩數之和等於目標數

LeetCode 兩數之和等於目標數

給定一個整數數列,找出其中和為特定值的那兩個數。

你可以假設每個輸入都只會有一種答案,同樣的元素不能被重用。

給定 nums = [2, 7, 11, 15], target = 9
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路是將陣列中的元素用HaspMap儲存,儲存的形式有點不一樣,由於最終需要的結果是下標。
所以我們需要將陣列中的元素作為key,元素的下標作為value,例如 map = {"2":0,"7":1,"11":2,"15":4} 

在遍歷存放的時候就要進行判斷(特殊情況如果有相同元素並且之和為目標結果)如果map中包含了該value,判斷該元素*2也就是當前元素和之前的元素之和為目標結果了,這時候就返回當前的下標和之前的已經存入map那個元素的下標,如果雖然

map中包含了該value,但是元素*2不等於目標數,就先put存放然後在continue判斷下一個元素,如果沒有包含在存放之前用目標結果數減去當前的數值得出的結果是否在map中已經存在,並且如果(某數*2 = 目標數)目標數減去自己還是這時候取下標還是自己,所以要加上判斷條件map.get(target-nums[i])!=i,如果存在就直接return兩個數的下標(還需要注意返回下標的順序),否則再put存放map中繼續遍歷。

HashMap<Integer, Integer> hMap = new HashMap<>();
		for (int i = 0; i < nums.length; i++) {
			int temp = nums[i];
			if (!hMap.containsKey(temp)) {
				hMap.put(temp, i);
				if (hMap.containsKey(target - temp) && hMap.get(target - temp) != i) {
				return new int[] { hMap.get(target - temp), i };
				}
			} else {
				if (temp * 2 == target) {
					return new int[] {hMap.get(temp),i};
				} else {
					hMap.put(temp, i);
					continue;
				}
			}
		}
return new int[] {};