home > algorithm > baekjoon > [baekjoon] 다음 소수 (백준 4134 java 풀이)

[baekjoon] 다음 소수 (백준 4134 java 풀이)
algorithm baekjoon step15

intro : 문제를 자세히 읽자, 크거나 같은 소수 중 가장 작은 소수를 찾는건데, 입력값 보다 큰 소수중 가장 작은 소수값을 찾으려고 삽질을 너무했다.

백준 문제링크

문제

정수 n(0 ≤ n ≤ 4*10^9)가 주어졌을 때, n보다 크거나 같은 소수 중 가장 작은 소수 찾는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다.

출력

각각의 테스트 케이스에 대해서 n보다 크거나 같은 소수 중 가장 작은 소수를 한 줄에 하나씩 출력한다.

문제 풀이

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int forCount = Integer.parseInt(br.readLine());
        for (int i = 0; i < forCount; i++) {
            long input = Long.parseLong(br.readLine());
            while (!isPrime(input)) {
                input++;
            }
            System.out.println(input);
        }
    }

    public static boolean isPrime(long input) {
        if (input <= 1) return false;
        for (long i = 2; i <= Math.sqrt(input); i++) {
            if (input % i == 0) {
                return false;
            }
        }
        return true;
    }
}