[파이썬 알고리즘 인터뷰] 7. 두 수의 합

 

 이 포스팅은 파이썬 알고리즘 인터뷰를 공부하면서 정리한 내용입니다.

 

 출처 : 파이썬 알고리즘 인터뷰

 코드 출처 : 파이썬 알고리즘 인터뷰 깃허브

 

---

7. 두 수의 합

리트코드 1번 문제입니다.

leetcode.com/problems/two-sum/

 

 

풀이

 딕셔너리를 활용한 풀이입니다. 키를 num, 값을 인덱스로 저장합니다. for 문으로 nums에서 num을 하나하나씩 꺼내보면서 딕셔너리에 target - num에 해당하는 키가 존재하는지 확인합니다.

def twoSum(self, nums, target):
    nums_map = {}
    for i, num in enumerate(nums):
        if target - num in nums_map:
            return [nums_map[target - num], i]
        nums_map[num] = i