[ leetcode ] Two Sum ( Java )

문제

leetcode Two Sum 풀이 ( 자바 )

 

 

풀이

간단히 두 수를 더하면 되는 것이므로 target - arr[i] 에 해당하는 키 값이 있는 지 확인해서 있다면 그 인덱스를 반환해주면 된다.

public int[] twoSum(int[] nums, int target) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int size = nums.length;
        
        for(int i = 0; i < size; i++) {
            int gap = target - nums[i];
            if(map.containsKey(gap)) {
                return new int[] {map.get(gap), i};
            }
            map.put(nums[i], i);
        }
        return null;
    }