코테 공부/백준

[Python] 1929 소수 구하기 *소수 판별법*

prefer_all 2022. 9. 14. 10:00
문제

M이상 N이하의 소수를 모두 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄에 자연수 M과 N이 빈 칸을 사이에 두고 주어진다. (1 ≤ M ≤ N ≤ 1,000,000) M이상 N이하의 소수가 하나 이상 있는 입력만 주어진다.

 

출력

한 줄에 하나씩, 증가하는 순서대로 소수를 출력한다.

 

예제 입력 예제 출력
3 16 3
5
7
11
13

풀이
# *** 오답: 시간 초과 ***
import sys 
input = sys.stdin.readline 

def check(i): #소수인지 검사
    for j in range(2, i):
        if i % j == 0:
            return False
    return True

M, N = map(int, input().split())
for i in range(M, N+1):
    if check(i) == True:
        print(i)

1. i를 2부터 i-1까지 나눠보며 검사하는 건

i를 2부터 i의 제곱근(루트 i)까지 나눠보며 검사하는 것과 동일

 

2. 1은 소수가 아니므로 따로 처리

import sys 
input = sys.stdin.readline 

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

M, N = map(int, input().split())
for i in range(M, N+1):
    if i == 1: # 1은 소수가 아니므로 제외
        continue
    if check(i) == True:
        print(i)