首页 > 代码库 > POJ 1087 A Plug for UNIX(网络流之最大流)
POJ 1087 A Plug for UNIX(网络流之最大流)
题目地址:POJ 1087
不知道是谁把这题化为了二分最大匹配的专题里。。于是也没多想就按照二分图的模型来建的(虽然当时觉得有点不大对。。。)。后来发现二分最大匹配显然不行。。有权值。。直接来个最大流多方便。。然后一直WA。。后来仔细想了想。。这根本就不能建二分图啊。。。。这题跟二分图一点关系都没有。。。。
这题的建图思路是让源点与每一个设备的插座类型连边,让汇点与每一个插座连边。然后用floyd判断该设备能否通过转换转换成可以插的插座上。只要可以转换成的就连边,权值为INF。然后求一次最大流,用n减去就行。
这题有个坑点,就是输入的插座可能会重复。。。需要去重。
代码如下:
#include <iostream> #include <cstdio> #include <string> #include <cstring> #include <stdlib.h> #include <math.h> #include <ctype.h> #include <queue> #include <map> #include <set> #include <algorithm> using namespace std; const int INF=0x3f3f3f3f; int m, plug[1100], dp[1100][1100]; int head[1100], source, sink, nv, cnt, n, tot, tt; int num[1100], d[1100], pre[1100], cur[1100]; struct node { int u, v, cap, next; } edge[1000000]; void add(int u, int v, int cap) { edge[cnt].v=v; edge[cnt].cap=cap; edge[cnt].next=head[u]; head[u]=cnt++; edge[cnt].v=u; edge[cnt].cap=0; edge[cnt].next=head[v]; head[v]=cnt++; } void bfs() { memset(d,-1,sizeof(d)); memset(num,0,sizeof(num)); queue<int>q; q.push(sink); d[sink]=0; num[0]=1; while(!q.empty()) { int u=q.front(); q.pop(); for(int i=head[u]; i!=-1; i=edge[i].next) { int v=edge[i].v; if(d[v]==-1) { d[v]=d[u]+1; num[d[v]]++; q.push(v); } } } } void isap() { memcpy(cur,head,sizeof(cur)); bfs(); int flow=0, u=pre[source]=source, i; while(d[source]<nv) { if(u==sink) { int f=INF, pos; for(i=source; i!=sink; i=edge[cur[i]].v) { if(f>edge[cur[i]].cap) { f=edge[cur[i]].cap; pos=i; } } for(i=source; i!=sink; i=edge[cur[i]].v) { edge[cur[i]].cap-=f; edge[cur[i]^1].cap+=f; } flow+=f; u=pos; } for(i=cur[u]; i!=-1; i=edge[i].next) { if(d[edge[i].v]+1==d[u]&&edge[i].cap) break; } if(i!=-1) { cur[u]=i; pre[edge[i].v]=u; u=edge[i].v; } else { if(--num[d[u]]==0) break; int mind=nv; for(i=head[u]; i!=-1; i=edge[i].next) { if(mind>d[edge[i].v]&&edge[i].cap) { mind=d[edge[i].v]; cur[u]=i; } } d[u]=mind+1; num[d[u]]++; u=pre[u]; } } printf("%d\n",n-flow); } void floyd() { int i, j, k; for(k=1; k<=tot; k++) { for(i=1; i<=tot; i++) { for(j=1; j<=tot; j++) { dp[i][j]=min(dp[i][k]+dp[k][j],dp[i][j]); } } } } int main() { int i, k, j; tot=0; tt=0; char s[30], s1[30]; map<string,int>q; scanf("%d",&m); memset(head,-1,sizeof(head)); cnt=0; for(i=1; i<=m; i++) { scanf("%s",s); if(!q[s]) q[s]=++tt; } tot=tt; scanf("%d",&n); for(i=1; i<=n; i++) { scanf("%s%s",s,s1); if(!q[s1]) q[s1]=++tot; plug[i]=q[s1]; } memset(dp,INF,sizeof(dp)); scanf("%d",&k); while(k--) { scanf("%s%s",s,s1); if(!q[s1]) q[s1]=++tot; if(!q[s]) q[s]=++tot; dp[q[s]][q[s1]]=1; //printf("--%d %d\n",q[s],q[s1]); } floyd(); /*for(i=1;i<=m;i++) { for(j=1;j<=m;j++) { printf("%d ",dp[i][j]); } printf("\n"); }*/ source=0; sink=tot+1; nv=sink+1; for(i=1; i<=n; i++) { add(source,plug[i],1); //printf("source--%d\n",plug[i]); for(j=1; j<=tt; j++) { if(dp[plug[i]][j]!=INF) { add(plug[i],j,INF); //printf("%d %d\n",plug[i],j); } } } for(i=1; i<=tt; i++) { add(i,sink,1); } //printf("%d %d\n",n,m); isap(); return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。