본문 바로가기
개발/알고리즘 문제

[Codility] CountDiv

by 유다110 2017. 11. 14.
반응형

0. 문제

세 개의 integer A, B, K 가 있다.

A와 B 사이에 있는 integer 중 K로 나누어 떨어질 수 있는 수의 개수를 구하여라.


1. 답변

문제는 쉬웠는데 예외 사항을 체크하지 못해 여러 번 돌려야 했다.

A나 B가 0일 경우엔 아무 수로나 나누어 떨어지는데 이걸 그냥 넘어감


A가 K 이상일 경우, B가 K 이상일 경우를 나눠 몫을 구하고,

A가 0일 경우엔 결과값에 1을 더했다.

def solution(A, B, K):
    result = 0
    if A >= K:
        result =  B // K - (A-1) // K
    elif B >= K:
        result =  B // K
    if A == 0:
        result += 1
    return result


2. 결과

반응형

댓글