---

백준 “소수 찾기” 문제를 풀어보았습니다.

여기를 누르면 건너뛸 수 있습니다.

---

문제

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

입력

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

출력

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

예제 입력 1

4
1 3 5 7

예제 출력 1

3

---

Solution

[문제 이해]

• 소수 찾기

이해한 내용을 바탕으로 코드를 작성했습니다.

[자바 코드]

package beakjoon;

import java.util.Scanner;

public class Main1978 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int T = sc.nextInt();
		int result = 0;
		for (int i=0; i < T; i++) {
			int num = sc.nextInt();
			if (isPrime(num)) result ++;
		}
		System.out.println(result);
		sc.close();
	}

	private static boolean isPrime(int num) {
		// TODO Auto-generated method stub
		if (num < 2) return false;
		for (int i=2; i*i <= num; i++) {
			if (num % i == 0) return false;
		}
		return true;
	}
}

[파이썬 코드]

import sys
input = sys.stdin.readline

# 라인 수를 줄인 코딩
def solution():
    N = int(input())
    is_prime = lambda x: 1 if sum([0 if x % i == 0 else 1 for i in range(2, int(x**0.5)+1 if int(x**0.5)+1 != 2 else 3)]) > 0 else 0 if x > 1 else 0
    result = sum([is_prime(x) for x in map(int, input().split())])
    print(result)

solution()