[Codility] Lesson5.CountDiv (C++)

🔮문제

Write a function:

int solution(int A, int B, int K);

that, given three integers A, B and K, returns the number of integers within the range [A..B] that are divisible by K, i.e.:

{ i : A ≤ i ≤ B, i mod K = 0 }

For example, for A = 6, B = 11 and K = 2, your function should return 3, because there are three numbers divisible by 2 within the range [6..11], namely 6, 8 and 10.

Write an efficient algorithm for the following assumptions:

• 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.

 

🔮풀이

1. A=B 일 경우

  A%K 가 0이면 1 리턴, 아니면 0을 리턴한다.

2. A != B일 경우

  (B를 K로 나눈 몫 - A를 K로 나눈 몫 + 1) 를 하면 그 사이에 K로 나눠지는 값들을 구할 수 있다.

---

🔮코드

 #include <algorithm>

int solution(int A, int B, int K) {
    if(A==B){
        if(A%K==0){
            return 1;
        }
        else{
            return 0;
        }
    }
    else{
        int start = 0;
        for(int i=A; i<=B; i++){
            if(i%K == 0){
                start = i;
                break;
            }
        }
        return B/K - start/K + 1;
    }
}