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

문제

 

주어진 수 N개 중에서 소수가 몇 개인지 찾아서 출력하는 프로그램을 작성하시오.

 

입력

 

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 

다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

 

출력

 

주어진 수들 중 소수의 개수를 출력한다.

 

풀이

 

스트림에 에라토스테네스의 체를 더해 풀어봤다.

 

주어진 숫자를 리스트로 받은 뒤 스트림을 열고,

 

그 안에서 다시 IntStream을 열어 2부터 √a 까지 순회하며 해당 순번의 숫자를 나눈 나머지가 0인 숫자를 필터링,

 

개수를 세어 출력했다.

 

시간이 너무 오래 걸리면 어쩌나 걱정했는데 스트림이 생각보다 느리지 않은 덕분인지

 

에라토스테네스의 체가 없이도 통과하기는 했다.

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

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

        int n = sc.nextInt();

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

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

        int result = (int) list.stream()
                .filter(a ->
                        a > 1 &&
                        IntStream.rangeClosed(2, (int)Math.sqrt(a))
                        .allMatch(b -> a % b != 0))
                .count();

        System.out.println(result);
    }
}