문제
- 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개 밖에 없다고 가정하겠습니다.
#1. 첫번째 코드
def two_sum(nums, target):
# 아래 코드를 작성해주세요.
for i in range(len(nums)):
for j in range(i+1, len(nums)):
if nums[i] + nums[j] == target:
return [i, j]
* 배운 것
- index 값을 출력하는 것인데 더해진 합계를 출력하는 것으로 문제를 잘못파악했다. 문제에 원하는 요지가 무엇인지 파악하는게 젤 중요!
- range([strat,] stop [,step]) 이 순서니까 까먹지 않는게 좋다!
#2. 두번째 코드
def test(a, b):
result = []
for i, v in enumerate(a):
temp = b - v
if temp in a:
result.append(i)
result.append(a.index(temp))
return result
* temp가 a에 존재하는가? 존재한다면 a의 index를 구하자!
* enumerate 함수
-> 자료형(list, set, tuple, dictionary, string)을 입력으로 받아 인덱스 값을 포함하는 enumerate 객체를 리턴. emerate(값)는 값의 인덱스 값을 얄려주는 것!