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

문제

X 대학 M 교수님은 프로그래밍 수업을 맡고 있다. 

 

교실엔 학생이 30명이 있는데, 학생 명부엔 각 학생별로 1번부터 30번까지 출석번호가 붙어 있다.

교수님이 내준 특별과제를 28명이 제출했는데, 그중에서 제출 안 한 학생 2명의 출석번호를 구하는 프로그램을 작성하시오.

 

입력

입력은 총 28줄로 각 제출자(학생)의 출석번호 n(1 ≤ n ≤ 30)가 한 줄에 하나씩 주어진다. 출석번호에 중복은 없다.

 

출력

출력은 2줄이다. 1번째 줄엔 제출하지 않은 학생의 출석번호 중 가장 작은 것을 출력하고, 2번째 줄에선 그다음 출석번호를 출력한다.

 

풀이

 

스트림으로 어떻게 풀지.. 하다가 며칠 전에 자바 8과 11을 비교한 포스팅을 적은 걸 기억해냈다.

 

거기서 얻은 Predicate키워드를 이용해 일단 1부터 30이 담긴 리스트를 하나 만들고,

 

문제에서 주어지는 숫자를 담을 크키 28의 리스트를 만들어 입력을 받아 또 하나의 리스트 2를 만들었다.

 

이후 스트림을 두 번 겹쳐 사용해 1부터 30까지 들어있는 리스트에서 리스트2와 겹치지 않는 숫자를 필터링했고,

 

오름차순으로 정렬한 후 하나씩 출력해 주었다.

 

코드로 보면 간단하다.

 

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.function.Predicate;
import java.util.stream.Collectors;

public class Prob5597Stream {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

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

        for (int i = 1; i <= 30; i++) {
            list.add(i);
        }

        for (int i = 0; i < 28; i++) {
            list2.add(sc.nextInt());
        }

        List<Integer> result = list.stream()
                .filter(a -> list2.stream().noneMatch(Predicate.isEqual(a)))
                .sorted().collect(Collectors.toList());

        System.out.println(result.get(0));
        System.out.println(result.get(1));
    }
}