본문 바로가기
자료구조&알고리즘/탐욕법

[이.코.테] Chapter 03 그리디 - 1이 될 때까지 (Java)

by _din 2020. 10. 30.

1이 될 때까지 문제 (교재 99p)

난이도 ●○○ | 시간 제한 1초 | 메모리 제한 128MB | 기출 2018 E 기업 알고리즘 대회

 

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

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

N과 K가 주어질 때 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 최소 횟수를 구하는 프로그램을 작성하시오.

 

입출력 조건)

입력 조건  - 첫째 줄에 N(2 ≤ N ≤ 100,000)과 K(2 ≤ K ≤ 100,000)가 공백으로 구분되며 각각 자연수로 주어진다. 이 때 입력으로 주어지는 N은 항상 K보다 크거나 같다.
출력 조건  첫째 줄에 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 최솟값을 출력한다.

입출력 예시)

입력 예시 출력 예시
25 5 2
17 4 3

 


 

문제 풀이 방법

  • 최대한 많이 나누기

 

소스 코드

package exam03_greedy;

import java.io.*;
import java.util.StringTokenizer;

public class Q4_MakeOne {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());

        int count = 0;
        while(n != 1) {
            if (n % k == 0) {
                n /= k;
            } else {
                n -= 1;
            }
            count ++;
        }

        bw.write(count+"\n");

        br.close();
        bw.close();

    }
}
반응형

댓글