백준 파이썬 1978 소수찾기

 

 

 

 

 

 

 

 

 

풀이

def p(n):
    a = 0
    b = 0
    while a<=n:
        a += 1
        if n%a==0:
            b += 1
        if a==n:
            if b==2:
                return 1
            else:
                return 0
 
a = int(input())
A = list(map(int,input().split()))
count = 0
for i in A:
    count += p(i)
print(count)

n이 소수인지 아닌지 판별하는 함수를 하나 만든다. 소수면 1, 소수가 아니면 0을 반환

그 안에다가 수들을 넣고 돌려서 카운팅