[Python] #1.Two Sum (Easy)

문제 #1. Two Sum

이 문제는 주어진 배열에 있는 두 개의 원소를 더해 target 값을 만들 수 있을때 그 두개의 index 값을 list로 반환하는 문제이다. 

처음 접근한 방법.

단순하게 2중 for문을 이용해서 작성했다. 

class Solution:
    def twoSum(self, nums,target):
        for i in range(len(nums)):
            for j in range(i+1,len(nums)):
                if nums[i] + nums[j] == target :
                    return [i,j]

문제 해결은 가능하나 역시 시간이 오래소요된다는 단점이 있다. 

따라서, for문을 한번만 활용하면서 문제를 해결할 수 있는 방법이 필요했고, dictionary를 이용해 문제를 해결하고자 했다. 

class Solution:
    def twoSum(self, nums,target):
        dic = {}
        for i,n in enumerate(nums):
            dif = target - n
            if dif not in dic:
                dic[n] = i
            else:
                return [dic[dif],i]

dictionary에 target과 현재의 값의 차이값이 없을 경우 현재의 값을 dictionary의 key값으로 index를 value값으로 입력한다. 

예를들어 nums의 값이 [3,2,4], target의 값이 6일 때,

dif 값 6-3 = 3 이고 dic[3]=0 이 입력된다.

dif 값 6-2 = 4, dic[2] = 1

dif 값 2 일 경우에 이전 case에서 dic[2] 값이 입력했기 때문에 해당 값의 index와 현재의 index값을 출력한다. 

이 방법을 사용하게 되면 for 문을 한번 만 사용하고 문제를 해결할 수 있다.