백준 1978번 소수찾기 (파이썬)

 

내 코드

n=int(input())

test=tuple(map(int,input().split()))

sosu=[]

for i in test:
    c=2
    while c<i:
        if i%c==0:
            break
        if c==i-1:
            sosu.append(i)
        c+=1

print(len(sosu))

내 첫코드

근데 틀렸다. 왤까...

2를 생각 못 하고 있었다...

 

n=int(input())

test=tuple(map(int,input().split()))

sosu=0

for i in test:
    if i==2:
        sosu+=1
    c=2
    while c<i:
        if i%c==0:
            break
        if c==i-1:
            sosu+=1
        c+=1

print(sosu)

2일때의 조건을 추가해주니 맞았다!!