Codeup 1274 : 소수 판별

https://codeup.kr/problem.php?id=1274

 

1274 : 소수 판별

# 1
n = int(input())
count = 0

for i in range(2, n+1):
    if n % i == 0:
        count += 1

if count > 2:
    print('not prime')
else:
    print('prime')
        

---

# 2
n = int(input())
now = 'prime'

for i in range(2, n):
    if n % i == 0:
        now = 'not prime'
        break

print(now)