[JAVA/백준] 1065번 - 한수 풀이

문제

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

입력

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

출력

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

풀이

import java.util.Scanner;
import java.util.stream.Stream;

public class Main {

	public static void main(String[] args) {
				
		Scanner scan = new Scanner(System.in);
		
		//최대 수
		int num = scan.nextInt();
		//한수 count 함수
		int han = han(num);
		
		System.out.println(han);
	}

	public static int han(int num) {
		
		int count = 0;
		
		//1부터 최대값까지 루프
		for(int i=1; i<=num; i++) {
		
			//각 자릿수의 숫자를 배열로 나누어 담음
			int[] arr = Stream.of(String.valueOf(i).split("")).mapToInt(Integer::parseInt).toArray();
			
			//두자리수까지는 무조건 한수 						
			if(arr.length == 1 || arr.length == 2) {
				count++;
			//세자리,네자리 수의 경우는 차이가 동일할 경우에만 count를 올림 
			} else if (arr.length == 3) {
				if((arr[0]-arr[1])==(arr[1]-arr[2])) {
					count++;
				} 
			} else if (arr.length == 4) {
				if((arr[0]-arr[1])==(arr[1]-arr[2]) && (arr[1]-arr[2])==(arr[2]-arr[3])) {
					count++;
				} 
			}
		}
		
		return count;
	}
}