hdu3526(最小割)

链接：点击打开链接

题意：有两个公司，每个公司有n种物品和每种物品的费用，给出m条指令，表示如果某两个物品不在一个公司被购买则需要再加额外的费用，问买n种不同的物品需要的最小费用

代码：#include <set>
#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
using namespace std;
const int INF=0x3f3f3f3f;
struct node{
int u,v,cap;
node(){}
node(int u,int v,int cap):u(u),v(v),cap(cap){}
}es[5000005];
int R,S,T;
int dis[1005],iter[1005];
vector<int> tab[1005];
void addedge(int u, int v, int cap){
tab[u].push_back(R);
es[R++]=node(u,v,cap);
tab[v].push_back(R);
es[R++]=node(v,u,0);
}
int bfs(){
int i,h;
queue<int> q;
q.push(S);
memset(dis,INF,sizeof(dis));
dis[S]=0;
while(q.size()){
h=q.front();
q.pop();
for(i=0;i<tab[h].size();i++){
node &e=es[tab[h][i]];
if(e.cap>0&&dis[e.v]==INF){
dis[e.v]=dis[h]+1;
q.push(e.v);
}
}
}
return dis[T]<INF;
}
int dfs(int x,int maxflow){
int flow;
if(x==T)
return maxflow;
for(int &i=iter[x];i<tab[x].size();i++){
node &e=es[tab[x][i]];
if(dis[e.v]==dis[x]+1&&e.cap>0){
flow=dfs(e.v,min(maxflow,e.cap));
if(flow){
e.cap-=flow;
es[tab[x][i]^1].cap+=flow;
return flow;
}
}
}
return 0;
}
int dinic(){
int ans,flow;
ans=0;
while(bfs()){
memset(iter,0,sizeof(iter));
while(flow=dfs(S,INF))
ans+=flow;
}
return ans;
} //网络流模板
int main(){
int n,m,i,j,u,v,num;
while(scanf("%d%d",&n,&m)!=EOF){
R=S=0,T=n+1;
for(i=0;i<=T;i++)
tab[i].clear();
for(i=1;i<=n;i++){ //求最小割，就可以满足
scanf("%d",&num); //用最小的费用使得两边至少选一个
addedge(0,i,num);
}
for(i=1;i<=n;i++){
scanf("%d",&num);
addedge(i,T,num);
}
for(i=1;i<=m;i++){ //在加费用时，只要加一个环就可以
scanf("%d%d%d",&u,&v,&num); //解决加费用的问题
addedge(u,v,num);
addedge(v,u,num);
}
printf("%d\n",dinic());
}
return 0;
}