codekata #1

codekata_0426

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

---

Need to consider

1. 2가지 숫자를 비교해서 더 해봐야 한다.
2. 같은 index를 가지면 안된다.

정답

def two_sum(nums, target):
    # 아래 코드를 작성해주세요.
    for i in range(len(nums)):					# index number을 하나씩 빼오자
        for j in range(len(nums)):				# index number을 하나씩 빼오자
            if i == j:						# i, j(index number임) 를 비교해서 인덱스가 같은 곳이면 
                continue					# 위에서 다시 돌리자
            elif i != j and nums[i]+nums[j] == target:		# 비교 후 i와 l이 다르고, nums[i]+nums[j]가 target과 같으면 
                return [i, j]					# i, j(index number임) 를 리스트로 뽑아내자
`
nums = [4, 4, 11, 14]
target = 15
print(two_sum(nums, target))

참고 내용

• range(), len(), for문, continue, 변수[index]