ZOJ3988 Prime Set （匈牙利算法求解）

题目链接：ZOJ - 3988 Prime Set

题意：

• 给出n个数， a[1]~a[n]
• 若a[i]+a[j]是素数，那么称 集合{i, j} 为Prime set
• 给定k， 表示至多从由这n个数组成的Prime set中取出k个集合
• 问：取出的集合合并后，集合的大小最大为多少？（注意，此处集合元素是下标，而不是a[i]，{1, 2} U {1, 3} = {1, 2, 3} ，故集合大小为3）

解法：

• 首先，求出所有的Prime Set集合
• 先从中选出 所有的两两不相交的集合， 假设其个数为t（如{1, 3}, {2, 4}，这就是求最大匹配数）
• 若 t >= k ，由于每个集合有两个元素所以，答案即为 2*k （因为至多选择出k个两两不相交的集合，而每个集合有两个元素，故为2* k）
• 若 t < k ，即取出所有两两不相交的集合后，还可选择 k-t个未被匹配的元素，先求出未被匹配的元素，假设其个数为 tt， 那么这答案就是 2*t + min(tt, k-t)，（因为建图时候，元素i都会与另外一个元素j相连表示{i, j}是Prime set， 若i未被匹配，表示{ i, j}中，j已匹配其他元素，故若选择{i, j}只会得到1个不同元素i）

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a; i<=b; ++i)
#define repp(i,a,b) for(int i=b; i>=a; --i)
#define mp make_pair