[파이썬 PYTHON · 자바 JAVA] LeetCode【Two sum】

Two Sum - LeetCode

---

파이썬 코드

class Solution(object):
    def twoSum(self, nums, target):

        dic = {}

        for i in range(len(nums)):
            search = target - nums[i]
            if search in dic:
                return [dic[search],i]
            else:
                dic[nums[i]] = i

        return []

 

nums 라는 배열이 주어지고 nums의 원소를 골라 target 값을 만들 수 있는 index를 구하는 문제였다.

 

딕셔너리 자료 구조를 사용해서 풀었다. 

 

찾아야 하는 값이 dic에 없으면 추가하고, 있다면 답을 반환하는 방식이다.

 

 

자바 코드

import java.util.*;
class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer,Integer> map = new HashMap<>();
        
        for(int i=0;i<nums.length;i++){
        
            int search = target-nums[i];
            
            if(map.containsKey(search)){
                return new int[]{map.get(search),i};
            }else{
                map.put(nums[i],i);
            }
        }
        
        return new int[]{};
    }
}

자바 코드도 같은 방식으로 짜면 된다.