[Python] 백준 1978번 : 소수 찾기

1978번: 소수 찾기

 

문제

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

 

 

입력

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다.

다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

 

 

출력

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

 

 

---

✔ Solution

N = int(input())
num_list = list(map(int, input().split()))
count = 0

for i in num_list:
    cnt = 0
    if i == 1:
        continue
    for k in range(2, i+1):
        if i % k == 0:
            cnt += 1
    if cnt == 1:
        count += 1
print(count)

 

 

✔ Solution 2

N = int(input())
num_list = list(map(int, input().split()))

def prime(n):
    if n == 1:
        return False
    elif n == 2:
        return True
    for i in range(2, n):
        if n % i == 0:
            return False
    return True

count = 0
for i in num_list:
    if prime(i):
        count += 1

print(count)