Algorithm/BOJ

백준 1978번 C++

poopooreum 2023. 8. 4. 17:38
반응형
백준 1978번 소수

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

1978번: 소수 찾기

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

www.acmicpc.net



정답 코드

#include<iostream>
using namespace std;
int find(int n);
int main() {
	int n;
	cin >> n;
	int sum = 0;
	for (int x = 0; x < n; x++) {
		int a;
		cin >> a;
		int res = find(a);
		sum += res;
	}
	cout << sum;

}
int find(int n) {
	int cnt = 0;
	if (n == 1)
		return 0;
	if (n == 2)
		return 1;
	else {
		for (int x = 1; x <= n; x++) {
			if (n % x == 0)
				cnt++;
		}
	}
	if (cnt == 2)
		return 1;
	else
		return 0;
}

반응형