(Codakata1) two_sum

문제

twoSum함수에 숫자 리스트와 '특정 수'를 인자로 넘기면,더해서 '특정 수'가 나오는 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개 밖에 없다고 가정하겠습니다.

My answer

• 이중 for문으로 진행해도 되나, 개인적으로 사용하는 데 제약이 있어 in과 index를 적극 활용함

def two_sum(nums, target):
    for i in nums:
        if target - i in nums :
            a = [nums.index(i), nums.index(target-i)]
            return a
            
 print(two_sum([25, 8, 11, 75], 100)
 >>> [0, 3]