코-딩/Leetcode

[Leetcode] 219. Contains Duplicate II

힞뚜루마뚜루 2023. 3. 26. 16:01
728x90

문제 : https://leetcode.com/problems/contains-duplicate-ii/description/

난이도 : Easy

 

풀이
class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        nDict = {}
      
        for i, n in enumerate(nums):
            if n in nDict and abs(i - nDict[n]) <= k:
              return True
            nDict[n]=i

        return False

 

dictionary 자료형을 하나 만들고
처음 배열을 순회하면서 그 위치에 해당하는 인덱스와 숫자를 서로 역할을 바꿔 저장한다.

계속 돌면서 조건을 만족한다면 True 반환!

728x90