ZOJ - 3988 Prime Set 匈牙利算法思想(最大匹配数)

题意：给定n个数，让你选k对数，每对数加和是素数，使得这些数对的下标组成的集合尽量大
思路：首先要找下标不重复的一些数对：例如 (1,2) (3,5) (4,7) 这三个就是不重复下标的素数对
可以运用匈牙利算法的思想，把不重复下标的最大匹配数找出来(如果需要，也可以把这些数对找出来，即f[]数组表示某个数的前驱结点)；
如果这些数对还不够k对的话，我们再找其他可以组成素数对的，但是还没有选的数

 

#include<bits/stdc++.h>

using namespace std;
typedef pair<int,int> P;

const int maxn = 3000 + 7;
const int maxd = 2e6 + 7;

int T, n, k;
int a[maxn];
int f[maxn];
bool p[maxd];
bool vis[maxn];

void prim() {
    memset(p, 1, sizeof p);
    p[0] =  p[1] = 0;
    for(int i = 2; i < maxd; ++i) {
        for(int j = i+i; j < maxd; j += i) {
            p[j] = 0;
        }
    }
}
vector<int> vec[maxn];
void init() {
    for(int i = 1; i <= n; ++i) {
        vec[i].clear();
        f[i] = -1;
    }
    for(int i = 1; i <= n; ++i) {
        for(int j = i+1; j <= n; ++j) {
            if(p[a[i]+a[j]]) {
                vec[i].push_back(j);
                vec[j].push_back(i);
                f[i] = f[j] = 0;
            }
        }
    }
}
bool dfs(int id) {
    vis[id] = 1;
    for(auto i : vec[id]) {
        if(!vis[i]) {
            vis[i] = 1;
            if(f[i] == 0 || dfs(f[i])) {
                f[i] = id;
                f[id] = i;
                return true;
            }
        }
    }
    return false;
}
void solve() {
    int ans = 0;
    // 找到原图中，没有重复下标的素数点对的最大匹配数
    // 此时f[] 数组储存相互匹配的值
    for(int i = 1; i <= n; ++i) {
        if(f[i] != 0) continue;
        memset(vis, 0, sizeof vis);
        if(dfs(i)) ans++;
    }
    if(ans >= k) {
        cout << k*2 << endl;
        return;
    }
    // 剩余可选的数，其f[] 数组值为0（表示可以跟其他数组成素数，并且不在最大匹配的数对中）
    int t = (k - ans); ans *= 2;
    for(int i = 1; i <= n; ++i) {
        if(t <= 0) break;
        if(f[i] == 0) {
            t--;
            ans++;
        }
    }
    cout << ans << endl;
}
int main() {
    prim(); // 素数表
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &k);
        for(int i = 1; i <= n; ++i) {
            scanf("%d", &a[i]);
        }
        init(); // 建图（非严格二分图）
        solve();
    }
    return 0;
}