[Codility] 5. CountDiv

문제 링크

CountDiv

문제 요약

차집합 문제.
주어진 범위 A~B(A<=B) 사이의 값 중 K로 나눴을 때 나머지가 0인 수의 개수를 구하라.

요구사항

A and B are integers within the range [0..2,000,000,000]
K is an integer within the range [1..2,000,000,000]
A ≤ B

코드

범위가 매우 넓으므로 for-loop는 다메다!!!!!!

차집합 개념을 가져가자

A를 잘 생각해보자...

예를들어 범위가 6~10이고, K가 2일 때...

A 집합에는 2,4,6 3개가 들어간다.
B 집합에는 2,4,6,8,10 5개가 들어간다.

여기서 각 집합의 끝값과 초기값이 되는 부분은 A다!

A가 K로 나누어떨어진다 -> (A<=B)이니, 당연히 B에도 포함된다 -> 교집합에 추가!

따라서 교집합 부분에 A가 포함이 되냐, 안 되냐의 여부가 개수를 정한다!!!

public func solution(_ A : Int, _ B : Int, _ K : Int) -> Int {
    var addedNum: Int = 0
    addedNum = A%K == 0 ? 1 : 0  
    return (B/K + addedNum) - A/K 
}