[알고리즘] Two Sum (java 풀이)

문제

given an array of integers return indices of the two numbers such that they add up to a specific target. You may assume that each input would hava exactly one solution, and you may not use the same element twice

 

Example

nums = [2,7,11,14] , target = 9,
answer : [0,1]
Because nums[0] + nums[1] = 2 + 7 = 9

 

 

풀이

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class TwoSum {

	static int[] nums = { 2, 8, 11, 21 };
	static int target = 10;

	public static void main(String[] args) {

		Map<Integer, Integer> map = new HashMap<>();
		int[] result = new int[2];

		for (int i = 0; i < nums.length; i++) {
			if (map.containsKey(nums[i])) {
				result[0] = map.get(nums[i]);
				result[1] = i;

			} else {
				map.put(target - nums[i], i);
			}
		}

		System.out.println(Arrays.toString(result));
	}
}

-두 수의 합이 10이되는 두 개의 배열 원소를 구하는 문제이다.

- (target - answer1) = (answer2) 가 되어야한다. 따라서 map에 answer2를 put 하고, 배열 index를 돌면서 그 원소가 나오는 인덱스를 구한다.