LeetCode/Python/2006.py

# 2006. Count Number of Pairs With Absolute Difference K

import sys
from typing import List


class Solution:
    def countKDifference(self, nums: List[int], k: int) -> int:
        r: int = 0
        i: int = 0
        l: int = len(nums)
        while i < l:
            j: int = i
            while j < l:
                if abs(nums[i] - nums[j]) == k:
                    r += 1
                    pass
                j += 1
                pass
            i += 1
            pass
        return r
    pass


def main() -> int:
    def r(ns: List[int], k: int) -> None:
        print(
            f"Solution().countKDifference({ns}, {k}) = {Solution().countKDifference(ns, k)}"
        )
        pass

    r([1, 2, 2, 1], 1)
    r([1, 3], 3)
    r([3, 2, 1, 5, 4], 2)
    return 0


if __name__ == "__main__":
    sys.exit(main())