문제

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

내 풀이

def two_sum(nums, target):				
    answer = []					# 정답을 출력시키는 리스트를 선언한다.
    for i in range(len(nums)):			# 배열안에 모든 숫자들이 서로 만나야 하기 때문에 이중 for문을 돌린다.
      for j in range(len(nums)):
        if nums[i] + nums[j] == target:		# 만약 배열안의 두 숫자가 target값과 같다면
          answer.append(i) , answer.append(j)	# answer 리스트안에 두개의 값을 넣는다.
    return(list(set(answer)))			# 그대로 출력 시키면 [0,1,1,0]이 나오기 때문에 set으로 묶어 중복을 제거 한 후 다시 리스트로 출력시켰다.

결과는 정답!!

계속 세션듣고 개념 이해하기만 하다가 알고리즘 문제를 풀이하니 기분이 새로웠다!