[BackJoon]1065번 스트림으로 풀기

문제

어떤 양의 정수 X의 각 자리가 등차수열을 이룬다면, 그 수를 한수라고 한다. 

등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다. 

N이 주어졌을 때, 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력하는 프로그램을 작성하시오. 

입력

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

출력

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

 

풀이

 

정의에 따라 1부터 99까지는 전부 한수이다. 또한 1000은 한수가 아니니 제외하면,

 

실제 문제는 '세 자리의 정수 중 각 자리의 차이가 같은 숫자'의 개수를 구하는 간단한 문제라는 걸 알 수 있다.

 

코드도 간단하다.

 

import java.util.Scanner;
import java.util.stream.IntStream;

public class Prob1065Stream {
    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt();

        if (n < 100) System.out.println(n);

        else {
            long result = IntStream.rangeClosed(100, n)
                    .filter(a -> (a / 100 - (a / 10) % 10) == ((a / 10) % 10 - a % 10))
                    .count();

            System.out.println(result + 99);
        }
    }
}