[알고리즘] 소수 판정

p = int(input()) # p에 값을 입력받는다
bool = True

if p <= 1: # 1보다 작거나 같으면 소수가 아니다
    bool = False
elif p > 2 and p % 2 == 0: # p가 2보다 크고 짝수면 소수가 아니다.
	bool = False
else:
    for i in range(3, int(p ** .5) + 1, 2): # 3부터 p의 제곱근까지 홀수만 나누어 본다
        if p % i == 0: 
            bool = False # p가 나누어떨어지면 소수가 아니다
            break
        
if bool:
    print('소수가 맞다')
else:
    print('소수가 아니다')

 

 

# 입력값
2

# 출력값
소수가 맞다