백준 - 2581. 소수

문제 설명

자연수 M과 N이 주어질 때 M이상 N이하의 자연수 중 소수인 것을 모두 골라 이들 소수의 합과 최솟값을 찾는 프로그램을 작성하시오.

예를 들어 M=60, N=100인 경우 60이상 100이하의 자연수 중 소수는 61, 67, 71, 73, 79, 83, 89, 97 총 8개가 있으므로, 이들 소수의 합은 620이고, 최솟값은 61이 된다.

입력

입력의 첫째 줄에 M이, 둘째 줄에 N이 주어진다.

M과 N은 10,000이하의 자연수이며, M은 N보다 작거나 같다.

출력

M이상 N이하의 자연수 중 소수인 것을 모두 찾아 첫째 줄에 그 합을, 둘째 줄에 그 중 최솟값을 출력한다.

단, M이상 N이하의 자연수 중 소수가 없을 경우는 첫째 줄에 -1을 출력한다.

 

import java.io.*;
import java.util.*;

public class Main {
    
    public static void main(String [] args)throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int a = Integer.parseInt(br.readLine());
        int b = Integer.parseInt(br.readLine());

        List<Integer> list = new ArrayList<>();
        int sum = 0;
        if(a==1)a++;
        for(int i = a ; i<=b ; i++){
            Boolean IsPrime = true;
            for(int j = 2; j<i ; j++){
                if(i%j == 0){
                    IsPrime = false;
                    break;
                }
            }
            if(IsPrime){
                list.add(i);
                sum+=i;
            }
        }
        if(list.size()>0){
            System.out.println(sum);
            System.out.println(list.get(0));
            
        }else{
            System.out.print(-1);
        }
    }
    
}

---

리팩토링

import java.io.*;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int a = Integer.parseInt(br.readLine());
        int b = Integer.parseInt(br.readLine());

        List<Integer> primeNumbers = new ArrayList<>();
        int sum = 0;

        for (int i = a; i <= b; i++) {
            if (isPrime(i)) {
                primeNumbers.add(i);
                sum += i;
            }
        }

        if (primeNumbers.size() > 0) {
            System.out.println(sum);
            System.out.println(primeNumbers.get(0));
        } else {
            System.out.println(-1);
        }
    }

    // 소수 판별 함수
    private static boolean isPrime(int n) {
        if (n < 2) return false; // 2 미만의 숫자는 소수가 아님
        for (int i = 2; i <= Math.sqrt(n); i++) { // √n까지만 확인
            if (n % i == 0) return false;
        }
        return true;
    }
}

 

1. 소수 판별하는 메서드 따로 뻬서 코드 간결성 높이기 

2. 소수 판별 시 제곱근까지만 확인하기(for문 횟수 줄이기) 

3. a가 1일때 연산 빼고 isPrime 메서드 안에서 2보다 작으면 return false ; 로 처리해주기