[TIL/Python] 두 수의 합

문제

배열이 주어졌을 때, 두 수를 더해서 특정 목표 값을 만들 수 있는 두 숫자의 인덱스를 반환하라.

예시: `nums = [2, 7, 11, 15]`, `target = 9`

출력: `[0, 1]` (2 + 7 = 9)

---

풀이

def two_sum(nums, target):
    # 변수 first_index는 nums 리스트의 각 인덱스를 순회하며 접근함
    for first_index in range(len(nums)):
        # second_index는 first_index 의 다음 인덱스부터 시작하여 끝까지 순회함
        for second_index in range(first_index + 1, len(nums)) :
            # 만약 num[first_index]의 값과 num[second_index]의 합이 target의 값과 같다면
            if nums[first_index] + nums[second_index] == target:
                # 두 인덱스를 리스트 형태로 반환함 (예: target: 9, num[0]: 2, num[1]: 7. 
                # 2 + 7 = 9
                return [first_index, second_index]
                
#예상 출력: [0, 1]
print(two_sum([2, 7, 11, 15], 9))