파이썬 소수 합성수 구하기

파이썬 소수 합성수 구하기

# 2 ~ 1000 까지 소수, 합성수 구하기
# 예) 997 입력
 
n = int(input())
cnt = 0
 
# 2부터 996 까지 나머지가 0 이면 prime
 
for i in range(2, n):
    if n % i == 0:
        cnt += 1
 
if cnt == 0:
    print("prime") # 소수 출력
else:
    print("composite") # 합성수 출력