Python/algorithm

[Leetcode] two-sum

baecode 2025. 9. 9. 02:37
반응형

Code

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        hash_1 = {}
        for i in range(len(nums)):
            if nums[i] in hash_1:
                return [i,hash_1[nums[i]]]
            comp = target - nums[i]
            hash_1[comp] = i
  • 시간 복잡도: O(n)

간단하게 푼다면 이중포문을 이용해서 O(n^2) 으로도 푼다.

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            for j in range(i+1, len(nums)):
                if nums[i] + nums[j] == target:
                    return [i, j]
  • 시간 복잡도: O(n^2)
반응형