[백준알고리즘] 1065 - 한수

www.acmicpc.net/problem/1065

1065번: 한수

 

이 문제는 수를 입력받아 그 수를 각 자릿수로 나눠 리스트로 만든 뒤, 바로 앞뒤 자릿수의 차를 계산하여 같은 경우 갯수(count)를 증가시키는 방식으로 문제를 해결할 수 있습니다.

 

이때, 1자리 또는 2자리 수의 경우는 차이가 없거나 하나이므로 등차인지 확인할 수 없으므로 자동으로 한수가 됩니다. 따라서 1자리 또는 2자리 수의 경우는 주어진 숫자를 그대로 반환하도록 하였습니다.

 

3자리 수 이상인 경우에만 각 자리수로 숫자를 나눠 리스트를 만든 뒤 앞뒤 자리수와의 차이를 비교하여 같은 경우 갯수를 증가시키고 다른 경우는 바로 반복을 종료하고 다음 수를 같은 방식으로 확인한 뒤, 최종적으로 count를 반환하도록 하였습니다.

 

자바 코드는 다음과 같습니다.

 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int num = Integer.parseInt(br.readLine());

        System.out.println(oneStep(num));
    }

    // 주어진 수보다 작거나 같은 한수의 개수를 찾아 반환하는 메소드
    private static int oneStep(int num) {
        int count = 99;

        // 한자리 또는 두자리 숫자의 경우 바로 그 수를 반환
        if (num < 100) {
            return num;
        }

        // 세자리 이상인 경우 주어진 수보다 작은 수을 반복하여 확인하면서
        for (int n = 100; n < num + 1; n++) {
            int a = n;

            // 각 자리수별로 수를 나눠 리스트로 만들기
            List<Integer> oneNums = new ArrayList<>();
            while (a != 0) {
                int val = a / 10;
                int remain = a % 10;
                oneNums.add(remain);
                a = val;
            }

            boolean result = true;  // 등차인지 확인
            int gap = oneNums.get(1) - oneNums.get(0);  // 맨 뒤 숫자와 그 앞에 있는 수의 차이 (기본 차이)
            // 각 자리수의 앞뒤 차이 구하기
            for (int i = 1; i < oneNums.size() - 1; i++) {
                int before = oneNums.get(i);
                int after = oneNums.get(i + 1);

                // 등차가 아닌 경우 반복 종료
                if (after - before != gap) {
                    result = false;
                    break;
                }
            }

            // 등차인 경우 한수 개수 증가
            if (result) {
                count++;
            }
        }

        return count;
    }
}