백준(BOJ) - 소수(2581번) - 파이썬(python)

문제

자연수 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을 출력한다.

 소수판별 함수 참고: https://jinhan0716.tistory.com/5

소수(prime number) 판별 코드 - 파이썬(python)

import sys
input=sys.stdin.readline

def is_prime(n):
    if n<2:
        return False
    if n in (2,3):
        return True
    if n%2==0 or n%3==0:
        return False
    if n<9:
        return True
    k,l=5,n**0.5
    while k<=l:
        if n%k==0 or n%(k+2)==0:
            return False
        k+=6
    return True

m=int(input().rstrip())
n=int(input().rstrip())

ans=0
min_v=n

for i in range(m,n+1):
    if is_prime(i):
        if ans==0:
            min_v=i
        ans+=i

if ans==0:
    print(-1)
else:
    print(ans)
    print(min_v)