PTA vector筛法求质数

此题用了埃拉托色尼筛法

本题要求使用筛法求出1~N以内的质数。
函数接口定义：
vector sieve(int n); //函数声明, 求n以内的质数

求n以内的质数。其中 n是传入的参数。n 的值不超过10 000 000的范围； 求出的质数存入容器vector并返回。
裁判测试程序样例：

#include <iostream>
#include <vector>
using namespace std;
vector<int> sieve(int n); //函数声明,求n以内的质数
int main(int argc, char const *argv[])

{

    int n;

    cin >> n;
   
    vector<int> ans = sieve(n);



    cout << ans.size() << endl;

    for (int i = 0; i < ans.size(); i++) {

        cout << ans[i];

        if (i < ans.size() - 1)cout << " ";

    }

    cout << endl;



    return 0;

}



/* 请在这里填写答案 */

输入样例：

输入在一行中给出一个正整数N，其值不超过10 000 000。

10

输出样例：

输出首先在一行中输出指定范围内的质数个数，然后在另一行输出指定范围内的所有质数，以空格分隔，但是最后一个质数后面没有多余空格。

4
2 3 5 7

 答案实现：

vector<int> sieve(int n){
    vector<int>ans;
	vector<bool> prime(n+1,1);					
	for(int i=2;i*i<=n;i++)					
	{
		if(prime[i])						 
		for(int j=i*2;j<=n;j+=i)			 
		{
			prime[j]=0;
		}
       }
       for(int k=2;k<=n;k++){
           if(prime[k])
               ans.push_back(k);
	}	
    return ans;
}//函数声明, 求n以内的质数