Code Kata(Python) week 1 - day 1

Jeongyun Heo·2021년 1월 25일
0
post-thumbnail

문제

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

문제풀이

중첩 for문 사용
if문 사용
리스트 함수 중에서 index 함수 사용

위치 반환(index)
index(x) 함수는 리스트에 x 값이 있으면 x의 위치 값을 돌려준다.
리스트 변수 이름 뒤에 .를 붙여서 리스트 관련 함수 사용

def two_sum(nums, target):
    for i in nums:
        for j in nums:
            if i + j == target:
                return [nums.index(i),nums.index(j)]

0개의 댓글