bzoj-4514(网络流)
题目链接:
4514: [Sdoi2016]数字配对
Description
有 n 种数字,第 i 种数字是 ai、有 bi 个,权值是 ci。
若两个数字 ai、aj 满足,ai 是 aj 的倍数,且 ai/aj 是一个质数,
那么这两个数字可以配对,并获得 ci×cj 的价值。
一个数字只能参与一次配对,可以不参与配对。
在获得的价值总和不小于 0 的前提下,求最多进行多少次配对。
Input
第一行一个整数 n。
第二行 n 个整数 a1、a2、……、an。
第三行 n 个整数 b1、b2、……、bn。
第四行 n 个整数 c1、c2、……、cn。
Output
一行一个数,最多进行多少次配对
Sample Input
3
2 4 8
2 200 7
-1 -2 1
2 4 8
2 200 7
-1 -2 1
Sample Output
4
HINT
n≤200,ai≤10^9,bi≤10^5,∣ci∣≤10^5
思路:
这是一个有限制条件的最大匹配问题,可以把每个点拆成左右两个点,左边与源点相连,右边与汇点相连,容量为b[i],费用为零,左右的一对点若满足题目要求的关系,
那么就连一对容量为无穷,费用为c[i]*c[j]的边,这两条边对称,最后建出来的图也是对称的,所以ans/2,在跑费用流的时候要用spfa沿最长路增广,判断费用和0的大小,就可以得到答案了;
AC代码:
#include <bits/stdc++.h> using namespace std; typedef long long LL; const int maxn=210; const LL inf=1e18; int n; LL A[maxn],B[maxn],C[maxn]; struct Edge { int from,to; LL cap,flow,cost; }; int m,s,t,inq[2*maxn],p[2*maxn]; LL d[2*maxn],a[2*maxn]; std::vector<Edge> edge; std::vector<int> G[2*maxn]; inline void add_edge(int from,int to,LL cap,LL cost) { edge.push_back((Edge){from,to,cap,0,cost}); edge.push_back((Edge){to,from,0,0,-cost}); m=edge.size(); G[from].push_back(m-2); G[to].push_back(m-1); } bool bellmanford(LL &flow,LL &cost) { for(int i=s;i<=t;i++)d[i]=-inf; memset(inq,0,sizeof(inq)); d[s]=0;inq[s]=1;p[s]=0;a[s]=inf; queue<int>qu; qu.push(s); while(!qu.empty()) { int fr=qu.front();qu.pop();inq[fr]=0; int len=G[fr].size(); for(int i=0;i<len;i++) { Edge& e=edge[G[fr][i]]; if(e.cap>e.flow&&d[e.to]<d[fr]+e.cost) { d[e.to]=d[fr]+e.cost; p[e.to]=G[fr][i]; a[e.to]=min(a[fr],e.cap-e.flow); if(!inq[e.to]){qu.push(e.to);inq[e.to]=1;} } } } if(d[t]<=-inf)return false; if(cost+d[t]*a[t]<0) { LL tep=cost/(-d[t]); flow+=tep; return false; } flow+=a[t]; cost+=d[t]*a[t]; int u=t; while(u!=s) { edge[p[u]].flow+=a[t]; edge[p[u]^1].flow-=a[t]; u=edge[p[u]].from; } return true; } inline LL mincostflow() { LL flow=0; LL cost=0; while(bellmanford(flow,cost)); return flow; } inline LL pow_mod(LL x,LL y,LL mod) { LL s=1,base=x; while(y) { if(y&1)s=s*base%mod; base=base*base%mod; y>>=1; } return s; } inline int isprime(LL num) { if(num<=1)return 0; if(num==2)return 1; if(num%2==0)return 0; for(int i=1;i<=50;i++) { LL x=rand()%num; if(x==0)x++; if(pow_mod(x,num-1,num)!=1)return 0; } return 1; } int main() { //freopen("in.txt","r",stdin); scanf("%d",&n); for(int i=1;i<=n;i++)scanf("%lld",&A[i]); for(int i=1;i<=n;i++)scanf("%lld",&B[i]); for(int i=1;i<=n;i++)scanf("%lld",&C[i]); for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) { if(A[i]/A[j]<=1||A[i]%A[j])continue; if(isprime(A[i]/A[j])) { add_edge(j,i+n,100000000,(LL)C[i]*C[j]); add_edge(i,j+n,100000000,(LL)C[i]*C[j]); } } } s=0,t=2*n+1; for(int i=1;i<=n;i++)add_edge(s,i,B[i],0),add_edge(n+i,t,B[i],0); printf("%lld\n",mincostflow()/2); return 0; }