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

문제

셀프 넘버는 1949년 인도 수학자 D.R. Kaprekar가 이름 붙였다. 

양의 정수 n에 대해서 d(n)을 n과 n의 각 자리수를 더하는 함수라고 정의하자. 

 

예를 들어, d(75) = 75+7+5 = 87이다.

양의 정수 n이 주어졌을 때, 이 수를 시작해서 n, d(n), d(d(n)), d(d(d(n))), ...과 같은 무한 수열을 만들 수 있다. 
예를 들어, 33으로 시작한다면 다음 수는 33 + 3 + 3 = 39이고, 그 다음 수는 39 + 3 + 9 = 51, 다음 수는 51 + 5 + 1 = 57이다. 이런식으로 다음과 같은 수열을 만들 수 있다.

33, 39, 51, 57, 69, 84, 96, 111, 114, 120, 123, 129, 141, ...

n을 d(n)의 생성자라고 한다. 위의 수열에서 33은 39의 생성자이고, 39는 51의 생성자, 51은 57의 생성자이다. 

생성자가 한 개보다 많은 경우도 있다. 예를 들어, 101은 생성자가 2개(91과 100) 있다. 

생성자가 없는 숫자를 셀프 넘버라고 한다. 100보다 작은 셀프 넘버는 총 13개가 있다. 1, 3, 5, 7, 9, 20, 31, 42, 53, 64, 75, 86, 97

10000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 출력하는 프로그램을 작성하시오.

출력

10,000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.

 

풀이

 

함수 파트에 있기도 하고, 셀프 넘버를 생성해 보라는 의도가 느껴지기도 하는 문제다.

 

하지만 그런거 없지 바로 스트림으로!

 

import java.util.ArrayList;
import java.util.List;
import java.util.function.Predicate;

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

        List<Integer> list1 = new ArrayList<>();
        List<Integer> list2 = new ArrayList<>();

        for (int i = 1; i <= 10000; i++) {
            list1.add(i);
        }

        for (int i = 1; i <= 10000; i++) {
            list2.add(i +
                    i % 10 +
                    (i % 100) / 10 +
                    (i % 1000) / 100 +
                    (i % 10000) / 1000);
        }

        list1.stream()
                .filter(a -> list2.stream()
                        .distinct()
                        .filter(b -> b <= 10000)
                        .noneMatch(Predicate.isEqual(a)))
                .sorted()
                .forEach(System.out::println);
    }
}

개인적으로는 5597번의 확장 버전이라는 느낌으로 풀었다.

 

1부터 10000을 가지고 있는 리스트 1을 만들고

 

적당한 범위의 셀프넘버가 아닌 수를 보유한 리스트 2를 만들어 스트림으로 중복제거, 비교, 정렬 후 하나씩 출력했다.