ALDS1_1_C: Prime Numbers

http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ALDS1_1_C&lang=jp
与えられた数の中に素数が何個あるか答える問題。
10000までの素数を求めておけばためし割りが少し早くなります。

#include<stdio.h>
#include<vector>
std::vector<int> ps;
void set_prime(int n){
	for(int i=2;i*i<=n;i++){
		if(n%i==0)return ;
	}
	ps.push_back(n);
}
bool is_prime(int n){
	if(n<2)return false;
	for(int i=0;i<ps.size();i++){
		int a=ps[i];
		if(a*a>n)return true;
		if(n%a==0)return false;
	}
	return true;
}

int main(){
	for(int i=2;i<10000;i++){
		set_prime(i);
	}
	int n,ans=0;
	scanf("%d",&n);
	while(n--){
		int a;
		scanf("%d",&a);
		ans+=is_prime(a)?1:0;
	}
	printf("%d\n",ans);
}
最終更新:2016年03月24日 02:21