[파이썬] 백준 4948 : 베르트랑 공준

[파이썬] 백준 4948 : 베르트랑 공준

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

4948번: 베르트랑 공준

실버 2

정수론, 소수 판별

---

접근

에라토스테네스의 체로 소수를 판별하는 방법을 사용하면 간단하게 구할 수 있다.

다만 내가 작성한 방법은 pypy3에서만 통과되었다.

 

코드

import sys; input = sys.stdin.readline
def isPrime(num):
        for i in range(2, int(num**0.5) + 1):
            if num % i == 0:
                return False
        return True

while True:
    n = int(input())
    cnt = 0

    if n == 0:
        break
    if n == 1:
        print(1)
    elif n > 1:
        for i in range(n+1, (2*n)):
            if isPrime(i):
                cnt += 1
        print(cnt)

 

1을 입력하면 그냥 1을 출력해버리고

2이상의 값은 소수를 판별하도록 만들었다.