ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • [이코테] 1이 될 때까지
    파이썬 2022. 1. 18. 16:21

    시간제한 : 1초 

    메모리 제한 : 128MB


    문제 :

    어떠한 수 N이 1이 될 때까지 다음의 두 과정 중 하나를 반복적으로 선택하여 수행하려고 한다. 단, 두 번째 연산은 N이 K로 나누어떨어질 때만 선택할 수 있다.

    1. N에서 1을 뺀다.
    2. N을 K로 나눈다.

    입력 조건 :

    첫째 줄에 N(2<=N<=100,000)과 K(2<=K<=100,000)가 공백으로 구분되며 각각 자연수로 주어진다. 이때 입력으로 주어지는 N은 항상 K보다 크거나 같다.


    출력 조건 :

    첫째 줄에 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 횟수의 최솟값을 출력한다.

     


    입력 예시 : 

    25 5

    출력 예시 : 

    2


    나의 풀이 :

    n,k=map(int,input().split())
    cnt=0
    while n>1:
        if n%k==0:
            cnt+=1
            n=n/k
        else:
            cnt+=1
            n-=1
    print(cnt)

    이렇게 푼다고 해서 지금 조건에서는 문제가 생기진 않는다. 하지만 만약 N이 더욱 큰 수의 범위를 갖게 되는 경우에도 코드가 빠르게 동작하기 위해서는 N이 K의 배수가 되도록 효율적으로 한 번에 만들어주는 코드를 작성해야 한다.

     

    답안 예시 : 

    n,k=map(int,input().split())
    result=0
    while True:
        target=(n//k)*k
        result+=(n-target)
        n=target
        if n<k:
            break
        result+=1
        n//=k
    result+=(n-1)
    print(result)

    (n//k)*k 표현이 다소 생소하다. 

    -1 연산을 반복해주는 것이 아닌 한번의 연산으로 K로 N이 나누어떨어질 수 있도록 만들어주는 연산이므로 코드의 속도가 훨씬 빨라질 수 있다. 반복문을 돌면서 -1을 빼주고 하는 것이 아닌 하나의 연산식을 통해 횟수를 만들어내는 표현을 배울 수 있었다.

     

    '파이썬' 카테고리의 다른 글

    [리트코드] 937 _ 로그 파일 재정렬  (0) 2022.01.12
    [리트코드] 125 _ 유효한 팰린드롬  (0) 2022.01.12
    [Python] 리스트  (0) 2022.01.10

    댓글

Designed by Tistory.