[CodeKata] 특정 배열에서 덧셈 조합찾기

 

two_sum 함수에 숫자 리스트와 '특정 수'를 인자로 넘기면, 더해서 '특정 수'가 나오는 index를 배열에 담아 return 해 주세요.

 

nums: 숫자 배열
target: 두 수를 더해서 나올 수 있는 합계
return: 두 수의 index를 가진 숫자 배열

 

예를 들어,

 

nums은 [4, 9, 11, 14]
target은 13 

nums[0] + nums[1] = 4 + 9 = 13 이죠?

그러면 [0, 1]이 return 되어야 합니다.

 

Supposition

 

target으로 보내는 합계의 조합은 배열 전체 중에 2개 밖에 없다고 가정하겠습니다.

 

Solution

def two_sum(nums, target):
    # 아래 코드를 작성해주세요.
    
   my_dict = {}
   
   for i in range(len(nums) - 1): 
     for j in range(i+1, len(nums)):
       my_dict[nums[i] + nums[j]] = [i, j]
       
   return my_dict[target]

 

딕셔너리변수를 빈 값으로 지정해주고 이중 for문을 돌려서 해결합니다.

 

첫 번째 반복문의 (nums)의 길이를 구해서 -1을 해준다 i 인자에 넣어줍니다.

(-1 넣은이유 out of range 오류 방지를 위해서이다. 마지막까지 더할 수 있기에)

 

두 번째 반복문에 j인자는 i를 제외한 숫자이기에 i인자에 +1을 해주고 처음부터 끝까지 길이를 구해 j 인자에 넣어준다.

 

그다음 빈 딕셔너리에 index [ i ]와 [ j ]를 키로 접근해서 더한 값을 list로 준다.

 

그리고 마지막 index [target]으로 반환해준다.