[백준][python][기본수학] 1978 소수 찾기 문제

1978번: 소수 찾기

✏️ 내가 작성한 코드 

n = int(input())
nlist = map(int, input().split())
cnt = 0

for num in nlist:
    if num == 1: # 1은 소수가 아니므로 continue
        continue
    else:
        for i in range(1,num+1):
            if i == num: # 밑에 else에서 break글 안거치고 끝까지 왔으므로 cnt 1증가
                cnt += 1
            else:
                if i != 1 and num%i == 0: # 1이 아닌 다른 수로 나눠지면 소수 아님
                    break
print(cnt)