[Algorithm] 백준 2581번 소수 (Python)

https://www.acmicpc.net/problem/2581

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

예제 입력 1

60
100

예제 출력 1

620
61

예제 입력 2

64
65

예제 출력 2

-1

풀이

설명

• 2부터 i까지 i가 j로 나누어 떨어지면 소수가 아니므로 cnt를 1 증가시키고 반복문을 빠져나온다.
• 만일 cnt의 값이 0이면 소수라는 의미이므로 arr에 해당 값을 넣는다.
• 소수가 없는 경우에는 arr가 비어있으므로 길이가 0이면 -1을 출력하고 아니면 arr의 합과 최솟값을 순서대로 출력한다.

코드

M = int(input())
N = int(input())
arr = []

for i in range(M, N+1):
    cnt = 0

    if i > 1:
        for j in range(2, i):
            if i % j == 0:
                cnt += 1
                break
        if cnt == 0:
            arr.append(i)

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