티스토리 뷰

알고리즘

백준 2581번 소수

killog 2020. 11. 11. 23:20
반응형

list filter 을 써본 기회였다.

map 이랑 쓰는 방법이 비슷한데, 참값만 filter 은 반환한다.

  • filter(함수, literable)
  • 두번째 인수인 반복 가능한 자료형 요소들을 첫번째 인자 함수에 하나씩 입력하여 리턴값이 참인 것만 묶어서 돌려준다.
  • 함수의 리턴 값은 참이나 거짓이어야 한다.

(참고 링크: wayhome25.github.io/cs/2017/04/03/cs-03/)


문제

자연수 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

def findPrime(n):
    if n==1:
        return 0
    for i in range(2, n):
        if n%i ==0:
            return False
    return n
M  = int(input())
N = int(input())
nums = list(range(M,N+1))
nums = list(filter(findPrime, nums))
if sum(nums)!=0:
    print(sum(nums))
    print(min(nums))
else:
    print(-1)
반응형
댓글