leetcode刷題之Two Sum(1)
阿新 • • 發佈:2018-12-24
給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那 兩個 整數,並返回他們的陣列下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個陣列中同樣的元素。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路
建立一個HashMap,key為陣列中的元素值,value為陣列下表,遍歷陣列,直到找到key為target - 陣列元素的鍵值對,時間複雜度O(n)
程式碼
Java
public class TwoSum {
public static int[] solution(int[] nums,int target){ Map<Integer,Integer> map=new HashMap<>(); int[] result=new int[2]; for(int i=0;i<nums.length;i++){ if(map.containsKey(target - nums[i])){ result[0]=map.get(target - nums[i]); result[1]=i; } map.put(nums[i],i); } return result; }
}