본문 바로가기
Python/코딩

배열 - 두수의 합

by Rainbound-IT 2021. 9. 9.
반응형

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

 

Two Sum - LeetCode

Level up your coding skills and quickly land a job. This is the best place to expand your knowledge and get prepared for your next interview.

leetcode.com

 

from typing import List


class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        nums_map = {}
        # 키와 값을 바꿔서 딕셔너리로 저장
        for i, num in enumerate(nums):
            nums_map[num] = i

        # 타겟에서 첫 번째 수를 뺀 결과를 키로 조회
        for i, num in enumerate(nums):
            if target - num in nums_map and i != nums_map[target - num]:
                return [i, nums_map[target - num]]

 

 

 

  
from typing import List


class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        nums_map = {}
        # 하나의 `for`문으로 통합
        for i, num in enumerate(nums):
            if target - num in nums_map:
                return [nums_map[target - num], i]
            nums_map[num] = i

 

 

 

반응형

'Python > 코딩' 카테고리의 다른 글

그룹 애너그램  (0) 2021.08.31
GitHub와 VSCode 연동하기  (0) 2021.07.29
main 함수  (0) 2021.06.22
'_' 언더스코어  (0) 2021.05.12
기억해야할 표현 - 파이썬알고리즘 인터뷰 공부  (0) 2021.05.06

댓글