[알고리즘 문제풀이] 코드카타 1

two_sum함수에 숫자 리스트와 '특정 수'를 인자로 넘기면, 더해서 '특정 수'가 나오는 index를 배열에 담아 return해 주세요.
(target으로 보내는 합계의 조합은 배열 전체 중에 2개 밖에 없다고 가정하겠습니다.)

• 입력: 정수 리스트 nums, 목표 숫자 target
• 출력: 두 수의 인덱스를 가진 숫자 리스트

풀이 과정

• 리스트의 원소와 해당 원소 바로 다음 원소의 합이 target과 같은지 검사한다.
• 같다면 결과 리스트에 추가한다.

코드

def two_sum(nums, target):
    # 아래 코드를 작성해주세요.
  result = []

  for i, element in enumerate(nums):
    for j in range(i+1, len(nums)) :
      if i+1 != len(nums) and element + nums[j] == target:
        result.append(i)
        result.append(j)
        break

  return result