[Leetcode]2006. Count Number of Pairs With Absolute Difference K

---

 

풀이

주어진 배열 nums 내 각각의 원소들의 차가 k 만큼인 pair의 수를 구하면 된다.

int로 초기화 된 dict를 선언한 후 배열 내 값에서 k를 뺴거나 더한 값을 각각 tmp,tmp2에 저장한다.

이후 선언해둔 dict에 차가 k 인 pair를 저장하여 그 수를 확인할 수 있게 된다.

 

코드

class Solution:
    def countKDifference(self, nums: List[int], k: int) -> int:
        dict = defaultdict(int)
        cnt = 0
        for num in nums:
            tmp, tmp2 = num+k, num-k
            if tmp in dict:
                cnt += dict[tmp]
            if tmp2 in dict:
                cnt += dict[tmp2]
            dict[num] += 1
        return cnt