[BOJ / Python] 2581 소수

문제

백준 2581 소수

코드

#BOJ 2581 소수

m = int(input())
n = int(input())

prime_num = []
for x in range(m, n+1):
    error = 0
    if x > 1:
        for i in range(2, x):
            if x % i == 0:
                error += 1
        if error == 0:
            prime_num.append(x)

if len(prime_num) == 0:
    print("-1")
else:
    print(sum(prime_num))
    print(min(prime_num))

코드 설명
소수를 찾는 코드는 백준 1978 소수 찾기에서 작성한 코드와 유사하다.
해당 문제에서는 소수의 합과 가장 작은 소수를 출력하라고 하였으므로 소수들을 prime_num 리스트에 넣어둔다.

주어진 범위 내에 소수가 없는 경우 prime_num 리스트 내에 아무것도 입력되지 않으므로 리스트의 길이가 0이다. 이것을 이용하여 len(prime_num) == 0 인 경우 -1이 출력되도록 한다.
그렇지 않은 경우에는 소수의 합과 최솟값을 차례대로 출력한다.