[Codility] Prefix Sums - CountDiv 풀이

예문

https://app.codility.com/programmers/lessons/5-prefix_sums/count_div/

Write a function:

class Solution { public 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.

해석

  • 세개의 정수 A,B,K 가 주어짐
  • { i : A ≤ i ≤ B, i mod K = 0 }
  • [A..B] 사이에 K로 나누어 떨어질수 있는 정수를 리턴하라!

풀이

  • A와 B사이에 K로 나눌수 있는 수 = (B / K) - (A / K)
  • A가 K로 나눌수 있는지 여부 = (A % K == 0)
  • A == B 이면 A와 B사이에 K로 나눌수 있는 수는 없음.

제약사항

  • A,B의 범위는 정수 [0..2,000,000,000]
  • K의 범위는 정수 [1..2,000,000,000]

코드

public int solution(int A, int B, int K) {
  int count = 0;
  if (A % K == 0) count++;
  if(A != B){
    int firstDivided = A / K;
    int secondDivided = B / K;
    count += secondDivided - firstDivided;
  }
  return count;
}

결과