Leetcode 2006 풀이(파이썬, 스위프트)

for문을 전체 길이-1 까지 돌리고 그 안에 2중 for문으로 i+1부터 전체길이까지

두 값의 차이가 k인 것들의 개수를 +해준다.

 

파이썬:

class Solution:
    def countKDifference(self, nums: List[int], k: int) -> int:
        cnt = 0 
        for i in range(len(nums)-1):
            for j in range(i+1, len(nums)):
                if abs(nums[i]-nums[j])==k:        # 차이의 절대값이 k일때
                    cnt+=1						   # cnt 1증가
        return cnt

스위프트:

class Solution {
    func countKDifference(_ nums: [Int], _ k: Int) -> Int {
        var cnt = 0
        for i in 0..<(nums.count-1){
            for j in (i+1)..<nums.count{
                if abs(nums[i] - nums[j]) == k{				//차이의 절대값이 k일때
                    cnt+=1		//cnt 1증가
                }
            }
        }
        return cnt
    }
}