【jzoj4832】【高维宇宙】【网络流】【二分图匹配】

题目大意

给出一个长度为n的序列a[i]，可以选两个数组成新的数，每个数最多使用一次，求最多组成多少个质数。

解题思路

首先算出有哪些数可以配对，用匈牙利求出最大匹配。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
#define LD double
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a>b)?b:a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const inf=2147483647;
int const maxn=80;
int n,gra,a[maxn+10],vis[maxn+10],heshu[2000+10],to[maxn*maxn+10],next[maxn*maxn+10],begin[maxn+10],pre[maxn+10];
void insert(int u,int v){
    to[++gra]=v;
    next[gra]=begin[u];
    begin[u]=gra;
}
bool dfs(int now){
    vis[now]=1;
    for(int i=begin[now];i;i=next[i])
        if((!pre[to[i]])||((!vis[pre[to[i]]])&&(dfs(pre[to[i]])))){
            pre[to[i]]=now;
            return 1;
        }
    return 0;
}
int main(){
    //freopen("prime.in","r",stdin);
    //freopen("prime.out","w",stdout);
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n)scanf("%d",&a[i]);
    fo(i,2,2000)
        if(!heshu[i])
            fo(j,2,2000/i)
                heshu[i*j]=1;
    fo(i,1,n-1)
        fo(j,i+1,n)
            if(a[i]&&a[j]&&(!heshu[a[i]+a[j]])){
                insert(i,j);
                insert(j,i);
            }
    int ans=0;
    fo(i,1,n){
        fo(j,1,n)vis[j]=0;
        if(dfs(i))ans++;
    }
    printf("%d",ans/2);
    return 0;
}