华为机试 HJ28 素数伴侣 -＞匈牙利算法+线筛法

华为机试 HJ28 素数伴侣

#include <bits/stdc++.h>
using namespace std;

const int N = 110, M = 10010, T = 7e4, D = 3e4 + 10;
vector<int> odds, ovens; //二分图存储奇数和偶数，只有奇数与偶数和才可能是素数
int n;
int h[D], e[M], ne[M], idx;
bool st[T];//i是否已经被最小质因子筛过了，也就是合数（线性筛)
int primes[T], cnt;//存储2-T之中的素数(质数)
// bool is_primes[T];//哈希表，i是否为素数
int match[D];//该偶数是否已被选中
int res;
bool st2[D];//表示为某个奇数选偶数时，标记是否已经考虑过该偶数

//邻接链表头插入
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

//线性筛质数
void get_primes(int x)
{
    for(int i = 2; i <= x; i ++){
        if(!st[i]) primes[cnt ++] = i;
        //然后用i乘以前面的素数筛后面的合数，且保证只被最小质因子筛掉
        for(int j = 0; primes[j] <= x / i; j ++){
            st[i * primes[j]] = true;
            if(i % primes[j] == 0) break;
        }
    }
}

//匈牙利算法
bool find(int x)
{
    for(int i = h[x]; i != -1; i = ne[i]){
        int oven = e[i];
        if(st2[oven]) continue;//已经被考虑过了不行
        st2[oven] = true;
        if(!match[oven] || find(match[oven])){
            match[oven] = x;
            return true;
        }
    }
    return false;
}

int main() {
    cin >> n;
    //头节点赋值-1,表示链式存储的null,也就是-1表示链表尾下一个
    memset(h, -1, sizeof h);
    int max_val = 0;
    for(int i = 0; i < n; i ++){
        int x;
        cin >> x;
        if(x & 1) odds.push_back(x);
        else ovens.push_back(x);
        max_val = max(max_val, x);
    }
    //线性筛出2-2*max_val的所有质因子
    get_primes(2 * max_val);
    //之后，奇数与偶数建表，临界链表存储边，以奇数为边头，偶数为边尾吧，如果和是素数就连起来
    for(auto a: odds)
        for(auto b: ovens)
            if(!st[a + b])
                add(a, b);
    //之后匈牙利算法
    for(auto x: odds){
        memset(st2, false, sizeof st2);//清空之前的标记状态
        if(find(x)) res ++;
    }
    cout << res << endl;
    return 0;
}
// 64 位输出请用 printf("%lld")