[LeetCode] Two Sum 파이썬 (dict, if ~ in)

풀이:

class Solution:
    def twoSum(self, nums, target):
        seen = {}
        for i, v in enumerate(nums):
            remaining = target - v
            if remaining in seen:
                return [seen[remaining], i]
            seen[v] = i
        return []

enumerate를 사용해서 nums 리스트를 순회한다.

target과 nums의 element를 뺀 나머지(remaining)가

seen의 key값으로 존재하는지 확인

존재한다면, seen의 remaininig 키값의 value와 현재 index(i)를 리스트의 원소로 각각 입력후 return

 

seen 이라는 dictionary를 만들어서 value를 key값으로, index를 value값으로 사용했다는 점이 상당히 깔끔.