백준 1978 - 소수 찾기

https://www.acmicpc.net/problem/1978

1978번: 소수 찾기

n = int(input())
arr = list(map(int, input().split()))
count = 0
for num in arr:
    flag = True
    if num==1:
        flag = False
    elif num!=1:
        for i in range(2, num):
            if num%i==0:
                flag = False
                break
    if flag==True:
        count+=flag
print(count)

값을 리스트로 입력받은 뒤 소수 여부를 판별하는 Flag를 설정하고, 소수에 해당하지 않는 경우에는 flag를 False로 바꾼 뒤 판별된 소수 여부를 count에 더해준다. True는 1이고 False는 0으로 연산되므로 반복문이 완료된 이후에는 count에 True로 표기된 경우의 합이 남는다.