[백준] 4948번_베르트랑공준

문제 링크 : https://www.acmicpc.net/problem/4948

4948번: 베르트랑 공준

def prime_num():
    nums = [i for i in range(2,123456*2+1)]
    prime_arr = []
    for num in nums:
        flag = True
        for i in range(2,int(num**0.5)+1):
            if num%i==0:
                flag = False
                break
        if flag:
            prime_arr.append(num)
    return prime_arr

arr = prime_num()                
while True:
    n = int(input())
    if n == 0 :
        break
    cnt=0
    for a in arr:
        if a>2*n:
            break
        if n<a<=2*n:
            cnt+=1
    print(cnt)

먼저 (123456*2)까지의 소수를 모두 구한 배열을 만든 후, 범위에 해당하는 소수를 더해야 시간초과가 안난다.

저 flag를 쓰기 싫어서

def isPrime(num):
    if num<2:
        return False
    for i in range(2,int(num**0.5)+1):
        if num%i==0:
            return False
    return True

def make_arr():
    prime_cnt = [0 for _ in range(123456*2+1)]
    nums = [i for i in range(1,123456*2+1)]
    for num in nums:
        if isPrime(num):
            prime_cnt[num] = 1
    return prime_cnt
            
arr = make_arr()              
while True:
    n = int(input())
    if n == 0 :
        break
    print(sum(arr[(n+1):(2*n+1)]))

이렇게도 풀어봤다.