[Python] 백준 1978번

1978번. 소수 찾기 (silver 4)

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

 

입력

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

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

 

출력

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

 

예제 입력 1

4
1 3 5 7

예제 출력 1

3

 

 

 

 

 

n = int(input()) # 입력받을 숫자의 개수
numbers = list(map(int, input().split(' '))) # 공백으로 숫자 구분
cnt = 0 

for i in numbers:
    prime = 0 
    if i == 1: # 1은 소수가 아니기 때문에 넘어간다
        continue
    for j in range(2, i+1):
        if i % j == 0:
            prime += 1  # prime == 1 이면 소수
    if prime == 1:
        cnt += 1        # 소수 개수 카운트
        
print(cnt)

 

input을 사용해서 입력받을 숫자의 개수를 n에 저장해준다.

그 다음 numbers에 공백을 두고 숫자를 입력받은 다음, 그 숫자를 list형태로 저장해준다.

numbers 안에 있는 원소들을 i라 하고 이에 대하여 for문을 사용한다.

prime의 기본값을 0 (소수가 아님)으로 저장해주고 원소가 1이라면 소수가 아니기 때문에 continue문을 사용해준다.

그 외의 경우 for문을 사용해서 2부터 i까지의 범위에서 i%j가 0인 경우, 나누어지는 수가 있으니 소수가 아니게 되므로 prime에 1을 더해준다.

마지막으로 for문 밖에 if문을 사용해서 prime == 1, 즉 소수라면 cnt에 1씩 더해서 소수 개수를 저장하게 해주었고, cnt를 출력해주었다.