DataScience trainee

Leetcode 1. Two Sum in Python3

Two Sum


개요

  • 이번 포스팅은 파이썬 코딩테스트 연습문제 풀이입니다.
  • Leetcode 에 공개된 예제이며 링크는 아래와 같습니다.

https://leetcode.com/problems/two-sum/


코드

def twoSum(self, nums: List[int], target: int) -> List[int]:
    # hash map을 사용하려고 합니다
    # nums에서 추출한 값을 x, target에서 x를 뺀 값을 y, 이 y를 기억해 둡니다.
    # x가 y와 일치되는 순간 탐색을 종료하고 index를 출력합니다.

    ys = {}

    for i, x in enumerate(nums):
        if x in ys:
            return [ys[x], i]

        y = target - x

        ys[y] = i

마치며

이번 포스팅은 Leetcode에 존재하는 난이도 Easy 에 해당하는 문제입니다.