首页 > 代码库 > ural 1076 Trash 二分图最大权匹配(费用流实现)
ural 1076 Trash 二分图最大权匹配(费用流实现)
统计每种垃圾的总和,若将K种垃圾倒入第F个垃圾桶,那么花费就是K-F(k) (自己已经有的垃圾不用倒)。
然后就是简单的二分图建图。
#include<cstdio> #include<queue> #include<algorithm> #include<cstring> using namespace std; #define MAXN 1000 #define MAXM 1000000 #define INF 0x3f3f3f3f struct node { int u,v,f,c,next; }e[MAXM]; int n,head[MAXN],pre[MAXN],dist[MAXN],vis[MAXN],ans; int en,s,t,maxflow,mincost; //s源点,t汇点 void add(int u,int v,int c,int f)//加边 { e[en].u=u; e[en].v=v; e[en].c=c; e[en].f=f; e[en].next=head[u]; head[u]=en++; e[en].u=v; e[en].v=u; e[en].c=-c; e[en].f=0; e[en].next=head[v]; head[v]=en++; } int spfa() { int i,u,v; for(i=0;i<=t;i++) pre[i]=-1,vis[i]=0,dist[i]=INF; dist[s]=0; vis[s]=1; queue<int>q; q.push(s); while(!q.empty()) { u=q.front(); q.pop(); for(i=head[u];i!=-1;i=e[i].next) { v=e[i].v; if(e[i].f>0&&dist[u]+e[i].c<dist[v]) { dist[v]=dist[u]+e[i].c; pre[v]=i; if(!vis[v]) { vis[v]=1; q.push(v); } } } vis[u]=0; } if(dist[t]==INF) return 0; return 1; } void add() { int v; int maxf=INF; for(v=pre[t];~v;v=pre[e[v].u]) maxf=min(maxf,e[v].f); for(v=pre[t];~v;v=pre[e[v].u]) { e[v].f-=maxf; e[v^1].f+=maxf; mincost+=e[v].c; } maxflow+=maxf; } void init() { maxflow=0; mincost=0; en=0; s=0; t=n+n+1; memset(head,-1,sizeof(head)); } int a[155][155]; int type[155]; int main() { while(scanf("%d",&n)!=EOF) { memset(type,0,sizeof(type)); for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) { scanf("%d",&a[i][j]); type[j]+=a[i][j]; } } init(); for(int i=1;i<=n;i++) add(s,i,0,1); for(int i=n+1;i<=n+n;i++) add(i,t,0,1); for(int i=1;i<=n;i++) { for(int j=n+1;j<=n+n;j++) { add(i,j,type[i]-a[j-n][i],1); } } while(spfa()) add(); printf("%d\n",mincost); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。