🍔 문제

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 되어야 합니다.

가정

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

🍟 풀이

#1 내 풀이

def two_sum(nums, target):
  for i in range(len(nums)):
    n = 1
    while i+n < len(nums):
      if nums[i] + nums[i+n] == target:
        return [i, i+n]
      else:
        n += 1

#2

def two_sum(nums, target):
  seen = {}

  for i, num in enumerate(nums):
    if target - num in seen:
      return [seen[target-num], i]
    else:
      seen[num] = i