[1065] 한수

[Silver IV] 한수 - 1065

문제 링크

성능 요약

메모리: 17164 KB, 시간: 200 ms

분류

브루트포스 알고리즘

문제 설명

어떤 양의 정수 X의 각 자리가 등차수열을 이룬다면, 그 수를 한수라고 한다. 등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다. N이 주어졌을 때, 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 1,000보다 작거나 같은 자연수 N이 주어진다.

출력

첫째 줄에 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력한다.

아이디어

어떤 수를 이루는 각 자리수가 등차수열임을 판별하는 문제였다.
이 문제에서 포인트는 숫자가 주어졌을 때, 각 자리수를 분해할 수 있는가였다.
나는 숫자를 문자열로 바꾼 뒤 split한 뒤 다시 정수형으로 형변환하는 방법을 선택했다.

답안코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int n = Integer.parseInt(br.readLine());

        int countHanNum = 0;
        for (int i = 1; i <= n; i++) {
            if (isHanNum(i)) countHanNum += 1;
        }

        System.out.println(countHanNum);
    }

    public static boolean isHanNum(int n) {
        // 1자리부터 2자리까지는 n에 값에 관계없이 각 자리수가 등차수열을 이룸
        if (n >= 1 && n <= 99) {
            return true;
        }

        // 문자의 각 자리 숫자로 이루어진 정수형 배열 선언
        int[] digit = Arrays.stream(String.valueOf(n).split(""))
                .mapToInt(Integer::parseInt)
                .toArray();

        // 차이가 같은지 비교
        int diff = digit[0] - digit[1];
        for (int i = 1; i < digit.length - 1; i++) {
            if (digit[i] - digit[i+1] != diff) {
                return false;
            }
        }

        return true;
    }
}

BOJ 카테고리 내 다른 글 보러가기