[LeetCode] Two Sum

같은수를 두번 사용하지 않고

 

입력받은 int형 배열에서 두개의 합이 target과 같은 배열의 인덱스를 찾는 알고리즘 문제이다.

 

ex)

 

Given nums = [2, 7, 11, 15], target = 9, Because nums[0] + nums[1] = 2 + 7 = 9, return [0, 1].

 

class Solution {
    public int[] twoSum(int[] nums, int target) {
 
        int[] result = new int[2];
 
        for(int i = 0; i < nums.length; i++){
            for(int j = 0; j < nums.length; j++){
                if(nums[i] + nums[j] == target && i != j){
                    result[0] = j;
                    result[1] = i;
                }
            }
        }
        return result;
    }
 
    public int[] twoSum1(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[] { i, map.get(complement) };
            }
        }
        throw new IllegalArgumentException("No two sum solution");
    }
 
    public int[] twoSum2(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i];
            if (map.containsKey(complement)) {
                return new int[] { map.get(complement), i };
            }
            map.put(nums[i], i);
        }
        throw new IllegalArgumentException("No two sum solution");
    }
}

 

 

twoSum(int[] nums, int target)

1. 반복문을 두번 중첩해서 하나하나 모두 확인하는 알고리즘이다.
2. BigO 표현법에 의해 배열의 길이 만큼 반복문을 두번 실행하기 때문에 n^2이 된다.

 

twoSum1(int[] nums, int target)

1. Hashmap에 key와 value로  int[]배열값을 하나씩 put한다.
2. map.containsKey(complement) 값을 이용하여 원하는 값이 있는지 확인한다.
3. value값이 i와 같지 않도록 하여 두수를 중복해서 사용하는 것을 방지한다.

twoSum2(int[] nums, int target)

1. Hashmap을 선언해 두고 원하는 값이 존재하지 않으면 현재 순서의 nums[i]를 put한다.(초기 Hashmap은 비어있다.)
2. 두번째 부터는 Hashmap에 값이 존재하기 때문에 Hashmap에 존재하는 값과 현재nums[i] 값을 계산하여 판단한다.(EX: complement = target - nums[2] , map에 complement key를 가지고 있는지 비교)
3. nums[i]의 값과 Hashmap 내부의 값은 서로 중복되지 않으므로 같은 두개의 값을 사용하지 않는다.

 

 

 

 

출처 : https://leetcode.com/problems/two-sum/