Jzoj4725 素数序列

题意：给你一个序列A，要求你求出A长度最大的子序列A'使得任意A'[i]+A'[j]都为素数

有一点是显然的，如果没有1，或者只有一个1，那么序列最长为2（一奇一偶），这种情况n^2枚举就好了

如果有多于一个1，那么我们可以选择所有的1，和最多一个偶数

注意会有长度为0和1的情况

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
bool vis[30000010];
int w[3000010],t,n,s[1010],m,c=0,a=0,b=0;
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		scanf("%d",s+i);
		m=max(m,s[i]); c+=(s[i]==1);
	}
	m<<=1;
	for(int i=2;i<=m;++i){
		if(!vis[i]) w[++t]=i;
		for(int j=1;j<=t&&i*w[j]<=m;++j){
			vis[i*w[j]]=1;
			if(i%w[j]==0) break;
		}
	}
	sort(s+1,s+1+n);
	if(c>1){
		for(int i=n;i;--i)
			if(!vis[s[i]+1]){
				printf("%d
",++c);
				for(;--c;printf("1 "));
				printf("%d",s[i]);
				return 0;
			}
		for(++c;--c;printf("1 "));
		return 0;
	}
	for(int i=1;i<=n;++i)
		for(int j=1;j<i;++j)
			if(!vis[s[i]+s[j]]&&s[i]+s[j]>s[a]+s[b]) a=i,b=j;
	if(!a && !b){
		for(int i=n;i;--i) 
			if(!vis[s[i]])  
			return 0&printf("1
%d",s[i]);
		puts("0");
	} else printf("2
%d %d",s[b],s[a]);
}