Codeforces 665D Simple Subset【构造】

题目链接:

http://codeforces.com/problemset/problem/665/D

题意:

给定序列,从中找出最大的子集,使得子集中的数两两相加均为质数。

分析:

貌似有用最大团做的。可是不会,名字太难听也不是很想学。
n 只有1000,暴力一发。
如果集合中有1的话,把所有1都放进去,我们最多再找一个偶数。
如果不考虑1的话,两个奇数,两个偶数相加均为偶数,所以最多找一个奇数和一个偶数。
枚举之后判断一下是否为质数就好了。
我觉得整个序列找不到两两相加为质数的时候的说明不是很清楚,wa了之后才知道找不到的话直接输出任何一个数。

代码:

#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
using namespace std;
vector<int>v1,v2,v, vv;
const int maxn = 1e3 + 5, maxm = 1e6+ 5;
int a[maxn];
bool isprime[2 * maxm];
void init()
{
    memset(isprime, true, sizeof(isprime));
    isprime[1] = false;
    for(int i = 2; i < 2 * maxm; i++){
        if(isprime[i]){
            for(int j = 2 * i; j < 2 * maxm; j += i){
                isprime[j] = false;
            }
        }
    }
}
int main (void)
{
    int n;
    scanf("%d", &n);
    bool flg = false;
    init();
    for(int i = 0; i < n; i++){
        scanf("%d", &a[i]);
        if(a[i] == 1){flg = true;v.push_back(1);}
        else if(a[i] & 1) v1.push_back(a[i]);
        else v2.push_back(a[i]);
    }
    if(flg){
        for(int i = 0; i < v2.size(); i++){
            if(isprime[v2[i] + 1]) {
                    v.push_back(v2[i]);
                    break;
            }
        }
    }
    bool flag = true;
    if(v.size() < 2){
        for(int i = 0; i < v1.size(); i++){
            for(int j = 0; j < v2.size(); j++){
                if(isprime[v1[i] + v2[j]]){ vv.push_back(v1[i]);vv.push_back(v2[j]);flag = false;break;}
            }
            if(!flag) break;
        }
    }
    if(v.size() == 0){v.push_back(a[0]);}
    if(vv.size() < v.size()){
        printf("%d\n", v.size());
        for(int i = 0; i < v.size(); i++) printf("%d ", v[i]);
    }else {
        printf("%d\n", vv.size());
        for(int i = 0; i < vv.size(); i++) printf("%d ", vv[i]);
    }
    return 0;
}

你可能感兴趣的:(Codeforces 665D Simple Subset【构造】)