본문 바로가기

Others/코딩 테스트

[Codility] Lesson 5 : CountDiv - JAVA

문제

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.
 
Copyright 2009–2022 by Codility Limited. All Rights Reserved. Unauthorized copying, publication or disclosure prohibited.

 

문제 해설

solution(int A, int B, int K)에 전달되는 A와 B의 범위는 0 ~ 2,000,000,000까지이고
K는 1 ~ 2,000,000,000까지이다.
A는 B보다 작거나 같다.

A와 B 사이에 있는 숫자들을 i라고 할 때 i는 A보다 크거나 같고, B보다 작거나 같다.
이때 i % K가 0인 숫자들의 개수를 구한다.

예를 들어 A = 6, B = 11, K = 2라고 할 때
6과 11 사이의 숫자들 중 2와 나누었을 때 나머지가 0이 되는 숫자는 6, 8, 10으로 3개의 숫자가 존재한다.
이 때 계산을 통해 찾아낸 숫자의 개수, 3을 반환한다.

 

풀이 - 1

class Solution {
    public int solution(int A, int B, int K) {
        int count = 0;
        for(int i=A; i<=B; i++) {
            if(i%K==0) count++;
        }
        return count;
    }
}

 

'풀이 - 1'에 대한 회고

 

가장 먼저 생각나는 대로 반복문 수행!

시간 복잡도가 O(n)이 나올걸 뻔히 알면서도 일단 해봤다.

 

 

역시나 퍼포먼스 0점이다.

 

 

그리고 시간복잡도는 O(B-A)로 나오긴 하지만 저게 O(n)이랑 엄청 차이나는 그런건 아닐듯 하다.

 

다른 사람의 풀이

class Solution {
    public int solution(int A, int B, int K) {
        if(A % K == 0) return (B/K) - (A/K) + 1;
        return (B/K) - (A/K);
    }
}

 

'다른 사람의 풀이'를 통한 회고

 

계속 반복문만 생각하다보니 답이 안나온다.

그래서 다른 사람의 풀이를 봤는데 2줄이면 끝난다.

역시 코딩테스트 연습을 많이 해본 사람들이라 다양한 방법을 잘 찾아!

'나는 아직 많이 부족하구나'라고 느끼고 있다.

 

 

정확도, 퍼포먼스는 당연히 100%

 

 

그리고 가장 중요한 시간복잡도!

어떻게든 시간복잡도를 줄여야된다고 생각했는데

반복문을 이용하면 무조건 O(n)이다.

그런데 다른 사람의 풀이는 딱 한 번만 계산하면 되니 O(1)!

엄청 효율적이다.