【PTA 7-7】求n以内最大的k个素数以及它们的和

7-7 求n以内最大的k个素数以及它们的和 （20 分)

本题要求计算并输出不超过n的最大的k个素数以及它们的和。

输入格式:

输入在一行中给出n(10≤n≤10000)和k(1≤k≤10)的值。

输出格式:

在一行中按下列格式输出:

素数1+素数2+…+素数k=总和值

其中素数按递减顺序输出。若n以内不够k个素数，则按实际个数输出。

输入样例1:

1000 10

输出样例1:

997+991+983+977+971+967+953+947+941+937=9664

输入样例2:

12 6

输出样例2:

11+7+5+3+2=28

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))

int a[1000005];

void prime()							  
{
	for(int i=2;i<=1000000;i++)
		a[i]=1;
	for(int i=2;i*i<=1000000;i++)
	{
		if(a[i]==1)
		{
			for(int j=i*i;j<=1000000;j+=i)
			{
				a[j]=0;
			}
		}
	}
}

int main()
{
	int n,m,i,flag=0,s=0,num=0;
	prime();
	cin>>n>>m;
	for(i=n;i>0;i--)
	{
		if(s==m)
			break;
		if(a[i])
		{
			if(flag)
				cout<<"+";
			cout<<i;
			flag=1;
			num+=i;
			s++;
		}
	}
	cout<<"="<<num<<endl;
	return 0; 
}