leetcode 兩數之和
阿新 • • 發佈:2019-02-04
給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。
你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。
示例:
給定 nums = [2, 7, 11, 15], target = 9 因為 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
class Solution { public int[] twoSum(int[] nums, int target) { Map<Integer,Integer> map = new HashMap<>(); for(int i=0;i<nums.length;i++) map.put(nums[i],i); for(int i=0;i<nums.length;i++){ int complement = target - nums[i]; if(map.containsKey(complement) && map.get(complement) != i){ return new int [] {map.get(complement),i}; } } throw new IllegalArgumentException("No two sum solution"); } }