[BOJ/백준] 1978번 소수 찾기

문제

주어진 수 N개 중에서 소수가 몇 개인지 찾아서 출력하는 프로그램을 작성하시오.

---

입력

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

---

출력

주어진 수들 중 소수의 개수를 출력한다.

 

입/출력

---

풀이

n = int(input()) #입력받을 개수 
arr = list(map(int,input().split()))#소수인지 판별할 배열 입력

def prime(num): #소수 판별 함수
    if num == 1:
        return False
    elif num == 2:
        return True
    for i in range(2,num):
        if num % i == 0:
            return False
    return True
cnt = 0
for i in arr:
    if prime(i):
        cnt += 1
print(cnt)

결과

---

코드 풀이

주석으로 대체