TIL - Code Kata

# 문제

• 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개 밖에 없다고 가정하겠습니다.

# 풀이

• 나의 풀이

def two_sum(nums, target):

    result = []

    for i in range(len(nums)):

      try:

        for j in range(i+1, len(nums)):

          if nums[i] + nums[j] == target:

            result.append(i)
            result.append(j)

            return result
          
          else:

            continue

      except IndexError:
        continue

• review
  • for문 중첩을 가장 먼저 생각했다. 주어진 리스트 배열에서 두 수를 골라야 하기 때문이다.
  • 더해서 target값과 일치할 수 있는 조합이 2개밖에 존재하지 않음이 의미하는 바는??
    1. 두 수는 서로 다르다.(두 수가 같으면 조합이 한 개밖에 존재하지 않기 때문이다.)
    2. 2개의 조합은 요소의 순서만 다를 뿐, 요소 값은 동일하다. ex) [1, 0], [0, 1]
  • 서로 다른 두 수의 조합을 찾되, 같은 요소로 이뤄진 조합의 중복을 피하기 위해 두 번째 for문에서 range()의 시작점을 i+1로 설정했다.
  • i+1로 인해 IndexError가 발생하므로 try, except로 처리했다.
  • pass vs continue
    • continue는 다음 루프로 이동하면서 아래 코드를 실행하지 않지만, pass는 현재 조건문만 실행하지 않고 아래 코드를 실행하게 된다.

• 다른 풀이

def two_sum(nums, target):

    answer = []

    for i in range(len(nums)):
      for j in range(len(nums)):
        if nums[i] + nums[j] == target:
          answer.append(i), answer.append(j)

    return list(set(answer)

• 내 풀이보다 훨씬 낫다는 생각이다. 코드의 길이가 짧은 것이 인상적이다.
• 내 코드와의 차이점은 for문 중첩으로 중복되는 2개의 조합을 모두 찾아냈다는 점이다. 그렇다보니, 찾은 요소값도 4개이다.
• 중복된 요소 제거를 set()을 사용하여 처리했다.