POJ 2391 Ombrophibic Bovines

求最少的时间能使牛都走到牛棚里，先求一次每对点间的最短路，然后二分答案

 

构图时，源点到每个牛棚i连一条容量为该牛棚初始牛数的边，对每个牛棚拆点，如果能从A走到B，连一条A到B"的边，容量为A初始的牛数，每个牛棚i，连一条道i"容量无穷大的边，最后每个点i"向汇点连一条容量为该牛棚能容纳的牛数的边，这样，保证从i到j"的牛不会从j跑到k,而从i到k在规定时间内无法到达

 

这道题卡了很久，老是WA，一开始以为是SAP模板错了，试了几次还不行，最后把floyd修改了下，先初始化，再赋值，最后过了，写floyd马虎不得，这上面吃了几次亏了

 

代码：

#include <cstdio> #include <cstring> #include <algorithm> #include <cmath> using namespace std; #define inf 1<<29 #define MAX 420 __int64 INF = 9999999999999; struct edge { int u,v; int c,f,op; int next; }; edge g[1000000]; int pre[MAX],dis[MAX],num[MAX],adj[MAX],cur[MAX],rc[MAX]; __int64 map[MAX][MAX],stack[MAX*MAX],res[MAX*MAX],w; int s,t,f,p,max_flow,tot,rn,vn,sum; int field[MAX],shelter[MAX]; void addedge(int u,int v,int c) { g[tot].u=u; g[tot].v=v; g[tot].c=c; g[tot].f=0; g[tot].next=adj[u]; adj[u]=tot++; g[tot].u=v; g[tot].v=u; g[tot].c=0; g[tot].f=0; g[tot].next=adj[v]; adj[v]=tot++; } void floyd() { int i,j,k; for(k=1;k<=f;k++) for(i=1;i<=f;i++) { if(map[i][k]<INF) { for(j=1;j<=f;j++) { if(map[k][j]<INF) { if(map[i][j]>map[i][k]+map[k][j]) map[i][j]=map[i][k]+map[k][j]; } } } } rn=0; for(i=1;i<=f;i++) for(j=i+1;j<=f;j++) { if(map[i][j]<INF) res[rn++]=map[i][j]; } sort(res,res+rn); } void sap() { int u,v,i; memset(dis,0,sizeof(dis)); memset(rc,0,sizeof(rc)); memset(num,0,sizeof(num)); for(u=0;u<=vn;u++) cur[u]=adj[u]; u=s; rc[s]=inf; num[0]=vn; while(dis[s]<vn) { for(i=cur[u];i!=-1;i=g[i].next) { if(g[i].c>0&&dis[u]==dis[g[i].v]+1) break; } if(i!=-1) { v=g[i].v; cur[u]=i; pre[v]=i; rc[v]=g[i].c-g[i].f; rc[v]=rc[v]<rc[u]?rc[v]:rc[u]; u=v; if(u==t) { for(;u!=s;u=g[pre[u]].u) { g[pre[u]].c-=rc[t]; g[pre[u]^1].c+=rc[t]; } max_flow+=rc[t]; rc[s]=inf; } } else { if(--num[dis[u]]==0) break; cur[u]=adj[u]; for(dis[u]=vn,i=adj[u];i!=-1;i=g[i].next) { if(g[i].c>0&&dis[u]>dis[g[i].v]) dis[u]=dis[g[i].v]; } dis[u]++; num[dis[u]]++; if(u!=s) u=g[pre[u]].u; } } } bool check(__int64 limit) { int i,j; memset(adj,-1,sizeof(adj)); tot=0; for(i=1;i<=f;i++) { addedge(s,i,field[i]); addedge(i,i+f,inf); addedge(i+f,t,shelter[i]); } for(i=1;i<=f;i++) for(j=i+1;j<=f;j++) { if(map[i][j]<=limit) { addedge(i,j+f,field[i]); addedge(j,i+f,field[j]); } } max_flow=0; sap(); if(max_flow==sum) return true; else return false; } void partition() { __int64 l=0,r=rn-1,mid,ans=-1; while(l<=r) { mid=(l+r)/2; if(check(res[mid])) { ans=mid; r=mid-1; } else l=mid+1; } if(ans==-1) printf("-1/n"); else printf("%I64d/n",res[ans]); } int main() { int i,j,a,b; while(scanf("%d %d",&f,&p)!=EOF) { sum=0; for(i=1;i<=f;i++) { scanf("%d %d",&field[i],&shelter[i]); sum+=field[i]; } for(i=1;i<=f;i++) { map[i][i]=0; for(j=1;j<=f;j++) map[i][j]=INF; } while(p--) { scanf("%d %d %I64d",&a,&b,&w); if(map[a][b]==-1||map[a][b]>w) { map[a][b]=map[b][a]=w; } } s=0; t=2*f+1; vn=t+1; floyd(); partition(); } //system("pause"); return 0; }