PAT 2021 春季 甲级 7-1 Arithmetic Progression of Primes

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 1e5+10;
int n,m,primes[maxn],num,vis[maxn];
void primes_table() { //质数表
	vis[0]=1,vis[1]=1;
	for(int i=2; i<=m; i++) {
		if(vis[i]!=0)continue;
		primes[num++]=i;
		for(int j=i+i; j<=m; j+=i) {
			vis[j]=1;
		}
	}
}
int main() {
	scanf("%d%d",&n,&m);
	//质数表
	primes_table();
	// 找到质数表中小于m的最大质数的位置
	if(num<n)printf("%d
",primes[num-1]); //没有找到数列，打印最后一个
	else if(num==n) {
		for(int i=0; i<num; i++) {
			if(i!=0)printf(" ");
			printf("%d",primes[i]);
		}
	} else { //num>n 寻找解决方案
		vector<int> tmp,anstmp;
		vector<vector<int>> ans;
		int mmaxd = -1; // 所有符合答案的序列中最大差 
		for(int i=0; i<num-1; i++) {
			int maxd=-1; // 以primes[i]为起点满足条件的序列的最大差 
			for(int j=i+1; j<num; j++) {
				int d=primes[j]-primes[i];
				if(d<maxd||d<mmaxd)continue;//剪枝，如果差小于已经找到的最大差，跳过 
				int c=primes[j];//数列第一个数
				tmp.clear(); //每次查找清楚tmp
				tmp.push_back(primes[i]); 
				while(c<m) { //在范围内
					if(vis[c]==1)break;
					tmp.push_back(c);
					c+=d;
					if(tmp.size()>n) { //如果有多个，选择第一个元素最大的
						tmp.erase(tmp.begin());
						anstmp=tmp;
						maxd=d; //更新已找到的等差质数序列的最大差 
					} else if(tmp.size()==n) {
						maxd=d; //更新已找到的等差质数序列的最大差 
						anstmp=tmp;
					}
				}
			}
			if(maxd>mmaxd)mmaxd=maxd;
		}
		if(anstmp.size()==0) {
			printf("%d",primes[num-1]);
		} else {
			for(int i=0; i<n; i++) {
				if(i!=0)printf(" ");
				printf("%d",anstmp[i]);
			}
		}
	}
	return 0;
}