[Python] 두인덱스의 합이 특정 숫자인 인덱스 구하기

문제

two_sum함수에 숫자 리스트와 '특정 수'를 인자로 넘기면, 더해서 '특정 수'가 나오는 index를 배열에 담아 return해 주세요.

nums: 숫자 배열
target: 두 수를 더해서 나올 수 있는 합계
return: 두 수의 index를 가진 숫자 배열

가정

target으로 보내는 합계의 조합은 배열 전체 중에 2개 밖에 없다고 가정하겠습니다.

이중 for문을 이용한 방법.

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]

딕셔너리를 이용한 방법.

def two_sum(nums, target):
    # 빈 딕셔너리 생성
    h = {}
    
    for i in range(len(nums)):
        # 찾아야 하는 인덱스의 값을 numberToFind에 저장 
        currentNumber = nums[i]
        numberToFind = target - currentNumber

        # h에 키가 numberToFind인 요소가 딕셔너리에 이미 있으면 [해당 키의 값, 현재i] 반환
        if (numberToFind in h):
            return [h[numberToFind], i]
        
        # h에 키가 numberToFind인 요소가 없으면 h에 '키(현재숫자): 값(인덱스)' 쌍 저장     
        h[currentNumber] = i
    return []

print(two_sum([1, 7, 11, 4], 8))
결과 [0, 1]
print(two_sum([4, 1, 11, 4], 8))
결과 [0, 3]

for문을 한번만 사용한 방법

아래 방법은 for문에서 index[i]해당 값을 가져와 target 넘버와 뺀 값을 기준으로 답은 찾는 방식이다.

def two_sum(nums, target):
  result=[]
  for i in nums:
    if target-i in nums:
        if i == target-i:
            result = [j for j in range(len(nums)) if nums[j]==i]
        else:
            result.append(nums.index(i))
            result.append(nums.index(target-i))
        return result