thiagowfx's avatar

Β¬ just serendipity πŸ€ (not just serendipity)

LeetCode #219: Contains Duplicate II

β€’ 54 words β€’ 1 min

LeetCode #219: Contains Duplicate II:

python
from collections import defaultdict

class Solution:
    def containsNearbyDuplicate(self, nums: List[int], k: int) -> bool:
        d = defaultdict(list)
        for i, num in enumerate(nums):
            d[num].append(i)

        for values in d.values():
            values.sort()
            # 1 3 4
            for a, b in zip(values[:-1], values[1:]):
                if abs(a - b) <= k:
                    return True

        return False