[Python] 백준 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 math
def is_prime(num):
    if num <= 1:
        return False
    for i in range(2, int(math.sqrt(num)+1)):
        if num % i == 0:
            return False
    return True

n = int(input())
m = int(input())
a = []
for x in range(n, m+1):
    if is_prime(x):
        a.append(x)

a.sort()
if sum(a) == 0:
    print('-1')
else:
    print(sum(a))
    print(a[0])

소수를 구하고 리스트안에 소수들을 집어넣어 sort를 해 sum과 첫번째 값(최솟값)을 구한다.

만약 sum이 0일 경우 소수가 없는 경우니깐 -1을 출력해준다.