【深基7.例2】质数筛——如何判断质数？

题目描述

输入 \(n\) 个不大于 \(10^5\) 的正整数。要求全部储存在数组中，去除掉不是质数的数字，依次输出剩余的质数。

输入格式

第一行输入一个正整数 \(n\)，表示整数个数。

第二行输入 \(n\) 个正整数 \(a_i\)，以空格隔开。

输出格式

输出一行，依次输出 \(a_i\) 中剩余的质数，以空格隔开。

样例 #1

样例输入 #1

5
3 4 5 6 7

样例输出 #1

3 5 7

提示

数据保证，\(1\le n\le100\)，\(1 \leq a_i \leq 10^5\)。

我的作答

#include <iostream>
#include <vector>
#include <cmath>
using namespace std;

bool isPrime(int x) {
	if (x<2) return 0;
	for (int j=2;j*j<=x;j++) {
		if (x % j ==0) {
			return 0;
		}
	}
	return 1;
}

int main() {
	int n,num;
	vector<int> a;
	vector<int> b;
	cin >> n;
	for (int i=0;i<n;i++) {
		cin >> num;
		a.push_back(num);
	}
	for (int j=0;j<a.size();j++) {
		if (isPrime(a[j])) {
			b.push_back(a[j]);
		}
	}
	for (int k=0;k<b.size();k++) {
		if (k==0) {
			cout << b[k];
		} else cout << " " << b[k];
	}
	return 0;
}

总结

关于如何判断质数

bool isPrime(int x) {
	if (x<2) return 0;
	for (int j=2;j*j<=x;j++) {
		if (x % j ==0) {
			return 0;
		}
	}
	return 1;
}

关于vector的使用

不要使用变量来作为数组长度，而应使用vector的push_pack方法在数组末尾添加元素、vector的size属性访问数组大小。