(二分最大流) 最大流 + 二分查找 小结

  做最大流题目的时候会遇到一种需要用二分查找的题型:

    (poj2455) 一张无向图中有 N 个点,M 条边,每条边都有一个权值,且每条边只能用一次,要求找出 T 条从 1 到 N 的路径,使这 T 条路径所经过的边中,权值的最大值最小。

  转化为最大流模型:T就是最大流,每条边只能用一次在网络流中就是容量为1。然后二分查找(枚举),边权小于或等于mid的边就加一条容量为1的网络流的边。最后知道最大流ans与T相等的最小的mid就是所求。

  在二分查找的时候,与一般的二分查找还是有区别的,一般不能有“ if(ans==T) break;” 这样的语句,因为这种语句使用的条件是函数严格递增,但是我们遇到的题目中很可能会有相邻的函数值相等的情况。

  一般情况下,二分写成这样:

    while(a<=b)
        {
            c=(a+b)/2;
             ……     
       if(ans>=sum) {res=c;b=c-1;}
            else a=c+1;
        }

  res就是我们所求的答案。

  

  其实这种题型的建图还是有套路可以找的,多做些题目就有思路了。

  hdu3081 Marriage Match II    

题目大意:
n个男孩n个女孩,女孩选男孩,每个女孩都要选到不同的人。吵过架的男孩女孩是不能匹配的,女孩也不能和自己的朋友吵过架的男孩匹配。
女孩每轮都选择没选过的男孩,问总共能选几轮。
题解:
女孩1..n,男孩n+1..2*n编号,由女孩到男孩建容量为1的边。起点s=2*n+1,到1..n建边,容量为mid,n+1..2*n到终点e=2*n+2建边,容量为mid。
二分搜索满足条件(n*mid==ans,ans是最大流)的mid,即为答案。
  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 using namespace std;
  6 const int N=220;
  7 const int M=200000, INF=0x3f3f3f3f;
  8 struct node
  9 {
 10     int to,next,w;
 11 }edge[M];
 12 int head[N],numh[N],h[N],cure[N],pre[N],f[N],vis[N][N];
 13 int ans,tot,mid;
 14 void SAP(int s, int e,int n)
 15 {
 16     int flow,u,tmp,neck,i;
 17     ans=0;
 18     memset(pre,-1,sizeof(pre));
 19     memset(h,0,sizeof(h));
 20     memset(numh,0,sizeof(numh));
 21     for(i=1;i<=n;i++)
 22         cure[i]=head[i];
 23     numh[0]=n;
 24     u=s;
 25     while(h[s]<n)
 26     {
 27         if(u==e)
 28         {
 29             flow =INF;
 30             for(i=s;i!=e;i=edge[cure[i]].to)
 31             {
 32                 if(flow>edge[cure[i]].w)
 33                 {
 34                     neck=i;
 35                     flow =edge[cure[i]].w;
 36                 }
 37             }
 38             for(i=s;i!=e;i=edge[cure[i]].to)
 39             {
 40                 tmp=cure[i];
 41                 edge[tmp].w-=flow;
 42                 edge[tmp^1].w+=flow;
 43             }
 44             ans+=flow;
 45             u=neck;
 46         }
 47         for(i=cure[u];i!=-1;i=edge[i].next)
 48             if(edge[i].w && h[u]==h[edge[i].to]+1) break;
 49         if(i!=-1) {cure[u]=i;pre[edge[i].to]=u;u=edge[i].to;}
 50         else
 51         {
 52             if(0==--numh[h[u]]) break; //GAPÓÅ»¯
 53             cure[u]=head[u];
 54             for(tmp=n,i=head[u];i!=-1;i=edge[i].next)
 55                 if(edge[i].w) tmp=min(tmp, h[edge[i].to]);
 56             h[u]=tmp+1;
 57             ++numh[h[u]];
 58             if(u!=s) u=pre[u];
 59         }
 60     }
 61 }
 62 void addedge(int i,int j,int w)
 63 {
 64     edge[tot].to=j;edge[tot].w=w;edge[tot].next=head[i];head[i]=tot++;
 65     edge[tot].to=i;edge[tot].w=0;edge[tot].next=head[j];head[j]=tot++;
 66 }
 67 int Find(int x){return x==f[x]?x:f[x]=Find(f[x]);}
 68 void Link(int x,int y)
 69 {
 70     int a=Find(x),b=Find(y);
 71     if(a!=b) {f[b]=a;}
 72 }
 73 int B[M],G[M];
 74 int main()
 75 {
 76     //freopen("test.txt","r",stdin);
 77     int cas,i,j,k,n,m,x,y,s,e,d,a,b;
 78     scanf("%d",&cas);
 79     while(cas--)
 80     {
 81         scanf("%d%d%d",&n,&m,&d);
 82         for(i=1;i<=n;i++) f[i]=i;
 83         s=2*n+1;e=s+1;
 84         for(k=1;k<=m;k++)
 85             scanf("%d%d",&G[k],&B[k]);
 86         while(d--)
 87         {
 88             scanf("%d%d",&x,&y);
 89             Link(x,y);
 90         }
 91         x=0,y=n;
 92         while(x<=y)
 93         {
 94             mid=(x+y)/2;
 95             tot=0;
 96             memset(head,-1,sizeof(head));
 97             memset(vis,0,sizeof(vis));
 98             for(i=1;i<=n;i++){
 99                 addedge(s,i,mid);
100                 addedge(i+n,e,mid);
101             }
102             for(i=1;i<=m;i++)
103             {
104                 for(j=1;j<=n;j++)
105                 {
106                     a=Find(G[i]),b=Find(j);
107                     if(a==b&&!vis[j][B[i]]) {addedge(j,B[i]+n,1);
108                     vis[j][B[i]]=1;}
109                 }
110             }
111             SAP(s,e,e);
112             if(ans>=n*mid) {d=mid;x=mid+1;}
113             else y=mid-1;
114         }
115         printf("%d\n",d);
116     }
117     return 0;
118 }
View Code

  

  hdu3277 Marriage Match III 与上面一题类似,多了拆点。拆点可以认为是把有N种属性的物品当为N个物品,但是他们之间还是有联系的。

  

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 using namespace std;
  6 
  7 const int N=800;
  8 const int M=1100000, INF=10000000;
  9 struct node
 10 {
 11     int to,next,w;
 12 }edge[M];
 13 int head[N],numh[N],h[N],cure[N],pre[N];
 14 int ans,n,tot;
 15 int f[N],G[M],B[M],mat[N][N];
 16 void SAP(int s, int e,int n)
 17 {
 18     int flow,u,tmp,neck,i;
 19     ans=0;
 20     memset(pre,-1,sizeof(pre));
 21     memset(h,0,sizeof(h));
 22     memset(numh,0,sizeof(numh));
 23     for(i=1;i<=n;i++)
 24         cure[i]=head[i];
 25     numh[0]=n;
 26     u=s;
 27     while(h[s]<n)
 28     {
 29         if(u==e)
 30         {
 31             flow =INF;
 32             for(i=s;i!=e;i=edge[cure[i]].to)
 33             {
 34                 if(flow>edge[cure[i]].w)
 35                 {
 36                     neck=i;
 37                     flow =edge[cure[i]].w;
 38                 }
 39             }
 40             for(i=s;i!=e;i=edge[cure[i]].to)
 41             {
 42                 tmp=cure[i];
 43                 edge[tmp].w-=flow;
 44                 edge[tmp^1].w+=flow;
 45             }
 46             ans+=flow;
 47             u=neck;
 48         }
 49         for(i=cure[u];i!=-1;i=edge[i].next)
 50             if(edge[i].w && h[u]==h[edge[i].to]+1) break;
 51         if(i!=-1) {cure[u]=i;pre[edge[i].to]=u;u=edge[i].to;}
 52         else
 53         {
 54             if(0==--numh[h[u]]) break; //GAPÓÅ»¯
 55             cure[u]=head[u];
 56             for(tmp=n,i=head[u];i!=-1;i=edge[i].next)
 57                 if(edge[i].w) tmp=min(tmp, h[edge[i].to]);
 58             h[u]=tmp+1;
 59             ++numh[h[u]];
 60             if(u!=s) u=pre[u];
 61         }
 62     }
 63 }
 64 void init()
 65 {
 66     for(int i=1;i<N;i++) f[i]=i;
 67     memset(mat,0,sizeof(mat));
 68 }
 69 void addedge(int i,int j,int w)
 70 {
 71     edge[tot].to=j;edge[tot].w=w;edge[tot].next=head[i];head[i]=tot++;
 72     edge[tot].to=i;edge[tot].w=0;edge[tot].next=head[j];head[j]=tot++;
 73 }
 74 int Find(int x){return x==f[x]?x:f[x]=Find(f[x]);}
 75 void Link(int x,int y)
 76 {
 77     int a=Find(x),b=Find(y);
 78     if(a!=b) f[b]=a;
 79 }
 80 int main()
 81 {
 82     //freopen("test.txt","r",stdin);
 83     int m,n,f,i,j,k,cas,s,e;
 84     scanf("%d",&cas);
 85     while(cas--)
 86     {
 87         scanf("%d%d%d%d",&n,&m,&k,&f);
 88         init();
 89         s=3*n+1; e=s+1;
 90         for(i=1;i<=m;i++)
 91         {
 92             scanf("%d%d",&G[i],&B[i]);
 93         }
 94         while(f--)
 95         {
 96             scanf("%d%d",&i,&j);
 97             Link(i,j);
 98         }
 99         for(i=1;i<=m;i++)
100         {
101             int g=Find(G[i]);
102             mat[g][B[i]]=1;
103         }
104         int a=0,b=n,mid;
105         while(a<=b)
106         {
107             mid=(a+b)/2;
108             tot=0;
109             memset(head,-1,sizeof(head));
110             for(i=1;i<=n;i++)
111             {
112                 addedge(s,i,mid);
113                 addedge(i,i+n,k);
114                 addedge(2*n+i,e,mid);
115             }
116             for(i=1;i<=n;i++)
117             {
118                 for(j=1;j<=n;j++)
119                 {
120                     if(mat[Find(i)][j]) addedge(i,j+2*n,1);
121                     else addedge(i+n,j+2*n,1);
122                 }
123             }
124             SAP(s,e,e);
125             if(ans>=n*mid) {a=mid+1;f=mid;}
126             else b=mid-1;
127         }
128         printf("%d\n",f);
129     }
130     return 0;
131 }
View Code

  

  poj2112  Optimal Milking 这种类型的基本题。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 using namespace std;
  6 const int N =250, M=100000,INF=0x3f3f3f3f;
  7 int Map[N][N], dist[N][N], pre[N];
  8 void floyd(int n)
  9 {
 10     int i,j,k;
 11     for(i=1;i<=n;i++)
 12     {
 13         for(j=1;j<=n;j++)
 14         {
 15             dist[i][j]=Map[i][j];
 16         }
 17     }
 18     for(k=1;k<=n;k++)
 19         for(i=1;i<=n;i++)
 20             for(j=1;j<=n;j++)
 21                 if(dist[i][k]!=INF&&dist[k][j]!=INF&&
 22                    dist[i][k]+dist[k][j]<dist[i][j])
 23                 {
 24                     dist[i][j]=dist[i][k]+ dist[k][j];
 25                 }
 26 }
 27 struct node
 28 {
 29     int to,next,w;
 30 }edge[M];
 31 int head[N],numh[N],h[N],cure[N];
 32 int ans,tot;
 33 void SAP(int s, int e,int n)
 34 {
 35     int flow,u,tmp,neck,i;
 36     ans=0;
 37     for(i=1;i<=n;i++)
 38         cure[i]=head[i];
 39     numh[0]=n;
 40     u=s;
 41     while(h[s]<n)
 42     {
 43         if(u==e)
 44         {
 45             flow =INF;
 46             for(i=s;i!=e;i=edge[cure[i]].to)
 47             {
 48                 if(flow>edge[cure[i]].w)
 49                 {
 50                     neck=i;
 51                     flow =edge[cure[i]].w;
 52                 }
 53             }
 54             for(i=s;i!=e;i=edge[cure[i]].to)
 55             {
 56                 tmp=cure[i];
 57                 edge[tmp].w-=flow;
 58                 edge[tmp^1].w+=flow;
 59             }
 60             ans+=flow;
 61             u=neck;
 62         }
 63         for(i=cure[u];i!=-1;i=edge[i].next)
 64             if(edge[i].w && h[u]==h[edge[i].to]+1) break;
 65         if(i!=-1) {cure[u]=i;pre[edge[i].to]=u;u=edge[i].to;}
 66         else
 67         {
 68             if(0==--numh[h[u]]) break; //GAPÓÅ»¯
 69             cure[u]=head[u];
 70             for(tmp=n,i=head[u];i!=-1;i=edge[i].next)
 71                 if(edge[i].w) tmp=min(tmp, h[edge[i].to]);
 72             h[u]=tmp+1;
 73             ++numh[h[u]];
 74             if(u!=s) u=pre[u];
 75         }
 76     }
 77 }
 78 void init()
 79 {
 80     tot=0;
 81     memset(head,-1,sizeof(head));
 82     memset(pre,-1,sizeof(pre));
 83     memset(h,0,sizeof(h));
 84     memset(numh,0,sizeof(numh));
 85 
 86 }
 87 void addedge(int i,int j,int w)
 88 {
 89     edge[tot].to=j;edge[tot].w=w;edge[tot].next=head[i];head[i]=tot++;
 90     edge[tot].to=i;edge[tot].w=0;edge[tot].next=head[j];head[j]=tot++;
 91 }
 92 int main()
 93 {
 94     //freopen("test.txt","r",stdin);
 95     int k,c,m,i,j,t,s,e,n,R,L,mid;
 96     while(scanf("%d%d%d",&k,&c,&m)!=EOF)
 97     {
 98         n=k+c;
 99         for(i=1;i<=n;i++)
100             for(j=1;j<=n;j++)
101             {
102                 Map[i][j]=INF;
103                 scanf("%d",&t);
104                 if(i==j) continue;
105                 if(t!=0) Map[i][j]=t;
106             }
107         floyd(n);
108         R=L=0;
109         for(i=1;i<=n;i++)
110             for(j=1;j<=n;j++)
111                 if(dist[i][j]!=INF) R=max(R,dist[i][j]);
112         s=n+1;e=s+1;
113         while(L<=R)
114         {
115             mid=(R+L)/2;
116             init();
117             for(i=1;i<=k;i++) addedge(s,i,m);
118             for(i=1;i<=c;i++) addedge(i+k,e,1);
119             for(i=1;i<=k;i++)
120                 for(j=k+1;j<=n;j++)
121                     if(dist[i][j]<=mid) addedge(i,j,1);
122             SAP(s,e,e);
123             if(ans>=c) R=mid-1,t=mid;
124             else L=mid+1;
125         }
126         printf("%d\n",t);
127     }
128     return 0;
129 }
View Code

 

  poj3228  Gold Transportation

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 using namespace std;
  6 
  7 const int N=410;
  8 const int M=100000, INF=0x3f3f3f3f;
  9 struct node
 10 {
 11     int to,next,w;
 12 }edge[M];
 13 int head[N],numh[N],h[N],cure[N],pre[N];
 14 int ans,tot;
 15 void SAP(int s, int e,int n)
 16 {
 17     int flow,u,tmp,neck,i;
 18     ans=0;
 19     for(i=1;i<=n;i++)
 20         cure[i]=head[i];
 21     numh[0]=n;
 22     u=s;
 23     while(h[s]<n)
 24     {
 25         if(u==e)
 26         {
 27             flow =INF;
 28             for(i=s;i!=e;i=edge[cure[i]].to)
 29             {
 30                 if(flow>edge[cure[i]].w)
 31                 {
 32                     neck=i;
 33                     flow =edge[cure[i]].w;
 34                 }
 35             }
 36             for(i=s;i!=e;i=edge[cure[i]].to)
 37             {
 38                 tmp=cure[i];
 39                 edge[tmp].w-=flow;
 40                 edge[tmp^1].w+=flow;
 41             }
 42             ans+=flow;
 43             u=neck;
 44         }
 45         for(i=cure[u];i!=-1;i=edge[i].next)
 46             if(edge[i].w && h[u]==h[edge[i].to]+1) break;
 47         if(i!=-1) {cure[u]=i;pre[edge[i].to]=u;u=edge[i].to;}
 48         else
 49         {
 50             if(0==--numh[h[u]]) break; //GAP优化
 51             cure[u]=head[u];
 52             for(tmp=n,i=head[u];i!=-1;i=edge[i].next)
 53                 if(edge[i].w) tmp=min(tmp, h[edge[i].to]);
 54             h[u]=tmp+1;
 55             ++numh[h[u]];
 56             if(u!=s) u=pre[u];
 57         }
 58     }
 59 }
 60 void init()
 61 {
 62     tot=0;
 63     memset(head,-1,sizeof(head));
 64     memset(pre,-1,sizeof(pre));
 65     memset(h,0,sizeof(h));
 66     memset(numh,0,sizeof(numh));
 67 
 68 }
 69 void addedge(int i,int j,int w)
 70 {
 71     edge[tot].to=j;edge[tot].w=w;edge[tot].next=head[i];head[i]=tot++;
 72     edge[tot].to=i;edge[tot].w=0;edge[tot].next=head[j];head[j]=tot++;
 73 }
 74 struct ND
 75 {
 76     int u,v,w;
 77 }e[M];
 78 int g[N],p[N];
 79 int main()
 80 {
 81     //freopen("test.txt","r",stdin);
 82     int n,m,i,j,k,s,E,a,b,c,sum,res,s1;
 83     while(scanf("%d",&n)!=EOF)
 84     {
 85         if(!n) break;
 86         sum=0;s1=0;
 87         for(i=1;i<=n;i++) {scanf("%d",&g[i]); sum+=g[i]; }
 88         for(i=1;i<=n;i++) {scanf("%d",&p[i]); s1+=p[i];}
 89         scanf("%d",&m);
 90         a=INF; b=0;
 91         int t=0;
 92         while(m--){
 93             scanf("%d%d%d",&i,&j,&k);
 94             e[t].u=i;e[t].v=j;e[t].w=k;t++;
 95             a=min(a,k); b=max(b,k);
 96         }
 97         if(sum>s1) {printf("No Solution\n");continue;}
 98         s=n+1; E=s+1;res=-1;
 99         while(a<=b)
100         {
101             c=(a+b)/2;
102             init();
103             for(i=1;i<=n;i++)
104             {
105                 if(g[i])addedge(s,i,g[i]);
106                 if(p[i])addedge(i,E,p[i]);
107             }
108             for(i=0;i<t;i++)
109                 if(e[i].w<=c) addedge(e[i].u,e[i].v,INF),addedge(e[i].v,e[i].u,INF);
110             SAP(s,E,E);
111             if(ans>=sum) {res=c;b=c-1;}
112             else a=c+1;
113         }
114         if(res==-1) printf("No Solution\n");
115         else printf("%d\n",res);
116     }
117     return 0;
118 }
View Code

 

  poj2455  Secret Milking Machine   

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 using namespace std;
  6 const int N=210;
  7 const int M=100000, INF=0x3f3f3f3f;
  8 struct node
  9 {
 10     int to,next,w;
 11 }edge[M];
 12 struct ND
 13 {
 14     int u,v,w;
 15 }e[40010];
 16 int head[N],numh[N],h[N],cure[N],pre[N];
 17 int ans,tot;
 18 void SAP(int s, int e,int n)
 19 {
 20     int flow,u,tmp,neck,i;
 21     ans=0;
 22     for(i=1;i<=n;i++)
 23         cure[i]=head[i];
 24     numh[0]=n;
 25     u=s;
 26     while(h[s]<n)
 27     {
 28         if(u==e)
 29         {
 30             flow =INF;
 31             for(i=s;i!=e;i=edge[cure[i]].to)
 32             {
 33                 if(flow>edge[cure[i]].w)
 34                 {
 35                     neck=i;
 36                     flow =edge[cure[i]].w;
 37                 }
 38             }
 39             for(i=s;i!=e;i=edge[cure[i]].to)
 40             {
 41                 tmp=cure[i];
 42                 edge[tmp].w-=flow;
 43                 edge[tmp^1].w+=flow;
 44             }
 45             ans+=flow;
 46             u=neck;
 47         }
 48         for(i=cure[u];i!=-1;i=edge[i].next)
 49             if(edge[i].w && h[u]==h[edge[i].to]+1) break;
 50         if(i!=-1) {cure[u]=i;pre[edge[i].to]=u;u=edge[i].to;}
 51         else
 52         {
 53             if(0==--numh[h[u]]) break; //GAP优化
 54             cure[u]=head[u];
 55             for(tmp=n,i=head[u];i!=-1;i=edge[i].next)
 56                 if(edge[i].w) tmp=min(tmp, h[edge[i].to]);
 57             h[u]=tmp+1;
 58             ++numh[h[u]];
 59             if(u!=s) u=pre[u];
 60         }
 61     }
 62 }
 63 void init()
 64 {
 65     tot=0;
 66     memset(head,-1,sizeof(head));
 67     memset(pre,-1,sizeof(pre));
 68     memset(h,0,sizeof(h));
 69     memset(numh,0,sizeof(numh));
 70 }
 71 void addedge(int i,int j,int w)
 72 {
 73     edge[tot].to=j;edge[tot].w=w;edge[tot].next=head[i];head[i]=tot++;
 74     edge[tot].to=i;edge[tot].w=w;edge[tot].next=head[j];head[j]=tot++;
 75 }
 76 bool  cmp(const ND &a, const ND &b)
 77 {
 78     return a.w<b.w;
 79 }
 80 int main()
 81 {
 82     //freopen("test.txt","r",stdin);
 83     int n,p,T,i,j,k,R,L,m,s;
 84     scanf("%d%d%d",&n,&p,&T);
 85     R=0;L=INF;
 86     for(i=0;i<p;i++)
 87     {
 88         scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
 89         k=e[i].w;
 90         R=max(R,k);
 91         L=min(L,k);
 92     }
 93     sort(e,e+p,cmp);
 94     while(L<=R)
 95     {
 96         init();
 97         m=(L+R)/2;
 98         for(i=0;i<p;i++)
 99         {
100             if(e[i].w>m)break;
101             addedge(e[i].u,e[i].v,1);
102         }
103         SAP(1,n,n);
104         if(ans>=T) {s=m; R=m-1;}
105         else L=m+1;
106     }
107     printf("%d\n",m);
108     return 0;
109 }
View Code

  

  poj2391  Ombrophobic Bovines 

  此题是这种类型中较难的题目,综合性挺高的。需要用floyd求任意两点之间的最短距离,然后二分查找。最烦的是超出了int范围,需要修改好几处。并且最不顺心的是习惯上用的二分查找在这里过不了。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 using namespace std;
  6 const int N =500,M=1000000;
  7 const long long INF=0x3f3f3f3f3f3f;
  8 long long  Map[N][N], dist[N][N],ans;
  9 void floyd(int n)
 10 {
 11     int i,j,k;
 12     for(i=1;i<=n;i++)
 13     {
 14         for(j=1;j<=n;j++)
 15         {
 16             dist[i][j]=Map[i][j];
 17         }
 18     }
 19     for(k=1;k<=n;k++)
 20         for(i=1;i<=n;i++)
 21             for(j=1;j<=n;j++)
 22                 if(dist[i][k]!=INF&&dist[k][j]!=INF&&
 23                    dist[i][k]+dist[k][j]<dist[i][j])
 24                 {
 25                     dist[i][j]=dist[i][k]+ dist[k][j];
 26                 }
 27 }
 28 struct node
 29 {
 30     int to,next,w;
 31 }edge[M];
 32 int head[N],numh[N],h[N],cure[N],pre[N];
 33 int tot;
 34 void SAP(int s, int e,int n)
 35 {
 36     int u,tmp,neck,i;
 37     long long flow;
 38     ans=0;
 39     for(i=1;i<=n;i++)
 40         cure[i]=head[i];
 41     numh[0]=n;
 42     u=s;
 43     while(h[s]<n)
 44     {
 45         if(u==e)
 46         {
 47             flow =INF;
 48             for(i=s;i!=e;i=edge[cure[i]].to)
 49             {
 50                 if(flow>edge[cure[i]].w)
 51                 {
 52                     neck=i;
 53                     flow =edge[cure[i]].w;
 54                 }
 55             }
 56             for(i=s;i!=e;i=edge[cure[i]].to)
 57             {
 58                 tmp=cure[i];
 59                 edge[tmp].w-=flow;
 60                 edge[tmp^1].w+=flow;
 61             }
 62             ans+=flow;
 63             u=neck;
 64         }
 65         for(i=cure[u];i!=-1;i=edge[i].next)
 66             if(edge[i].w && h[u]==h[edge[i].to]+1) break;
 67         if(i!=-1) {cure[u]=i;pre[edge[i].to]=u;u=edge[i].to;}
 68         else
 69         {
 70             if(0==--numh[h[u]]) break; //GAP优化
 71             cure[u]=head[u];
 72             for(tmp=n,i=head[u];i!=-1;i=edge[i].next)
 73                 if(edge[i].w) tmp=min(tmp, h[edge[i].to]);
 74             h[u]=tmp+1;
 75             ++numh[h[u]];
 76             if(u!=s) u=pre[u];
 77         }
 78     }
 79 }
 80 void init()
 81 {
 82     tot=0;
 83     memset(head,-1,sizeof(head));
 84     memset(pre,-1,sizeof(pre));
 85     memset(h,0,sizeof(h));
 86     memset(numh,0,sizeof(numh));
 87 
 88 }
 89 void addedge(int i,int j,int w)
 90 {
 91     edge[tot].to=j;edge[tot].w=w;edge[tot].next=head[i];head[i]=tot++;
 92     edge[tot].to=i;edge[tot].w=0;edge[tot].next=head[j];head[j]=tot++;
 93 }
 94 int c[N],x[N];
 95 int main()
 96 {
 97     //freopen("test.txt","r",stdin);
 98     int n,m,i,j,k;
 99     long long res,L,R,mid,s1,s2,w;
100     while(scanf("%d%d",&n,&m)!=EOF){
101         for(i=1;i<=n;i++)
102         {
103             Map[i][i]=0;
104             for(j=i+1;j<=n;j++)
105                 Map[j][i]=Map[i][j]=INF;
106         }
107         s1=s2=0;
108         for(i=1;i<=n;i++)
109         {
110             scanf("%d%d",&x[i],&c[i]);
111             s1+=x[i];s2+=c[i];
112         }
113 
114         while(m--)
115         {
116             scanf("%d%d%I64d",&i,&j,&w);
117             Map[i][j]=Map[j][i]=min(w,Map[i][j]);
118         }
119         if(s1>s2) {printf("-1\n");continue;}
120         for(i=1;i<=n;i++)
121             if(x[i]>c[i]) break;
122         if(i>n) {printf("0\n");continue;}
123         floyd(n);
124         int s=2*n+1,e=s+1;
125         L=0,res=-1,R=INF;
126         while(L<R)
127         {
128             init();
129             mid=(R+L)/2;
130             for(i=1;i<=n;i++)
131             {
132                 addedge(i,i+n,INF);
133                 addedge(s,i,x[i]);
134                 addedge(i+n,e,c[i]);
135                 for(j=1;j<=n;j++)
136                 {
137                     if(i==j) continue;
138                     if(dist[i][j]<=mid) addedge(i,j+n,INF);
139                 }
140             }
141             SAP(s,e,e);
142             if(ans>=s1) {res=mid; R=mid;}
143             else L=mid+1;
144         }
145         printf("%I64d\n",res);
146     }
147     return 0;
148 }
View Code

 

我的代码中最大流算法都是用的优化的SAP算法,因为他的时间复杂度低。并且我已经习惯了这代码。

SAP模版:

 1 const int N=110;
 2 const int M=2*N*N, INF=0x3f3f3f3f;
 3 struct node
 4 {
 5     int to,next,w;
 6 }edge[M];
 7 int head[N],numh[N],h[N],cure[N],pre[N];
 8 int ans,tot;
 9 void SAP(int s, int e,int n)
10 {
11     int flow,u,tmp,neck,i;
12     ans=0;
13     for(i=1;i<=n;i++)
14         cure[i]=head[i];
15     numh[0]=n;
16     u=s;
17     while(h[s]<n)
18     {
19         if(u==e)
20         {
21             flow =INF;
22             for(i=s;i!=e;i=edge[cure[i]].to)
23             {
24                 if(flow>edge[cure[i]].w)
25                 {
26                     neck=i;
27                     flow =edge[cure[i]].w;
28                 }
29             }
30             for(i=s;i!=e;i=edge[cure[i]].to)
31             {
32                 tmp=cure[i];
33                 edge[tmp].w-=flow;
34                 edge[tmp^1].w+=flow;
35             }
36             ans+=flow;
37             u=neck;
38         }
39         for(i=cure[u];i!=-1;i=edge[i].next)
40             if(edge[i].w && h[u]==h[edge[i].to]+1) break;
41         if(i!=-1) {cure[u]=i;pre[edge[i].to]=u;u=edge[i].to;}
42         else
43         {
44             if(0==--numh[h[u]]) break; //GAP优化
45             cure[u]=head[u];
46             for(tmp=n,i=head[u];i!=-1;i=edge[i].next)
47                 if(edge[i].w) tmp=min(tmp, h[edge[i].to]);
48             h[u]=tmp+1;
49             ++numh[h[u]];
50             if(u!=s) u=pre[u];
51         }
52     }
53 }
54 void init()
55 {
56     tot=0;
57     memset(head,-1,sizeof(head));
58     memset(pre,-1,sizeof(pre));
59     memset(h,0,sizeof(h));
60     memset(numh,0,sizeof(numh));
61 
62 }
63 void addedge(int i,int j,int w)
64 {
65     edge[tot].to=j;edge[tot].w=w;edge[tot].next=head[i];head[i]=tot++;
66     edge[tot].to=i;edge[tot].w=0;edge[tot].next=head[j];head[j]=tot++;
67 }
View Code

 

posted @ 2014-09-17 01:25  pengmq  阅读(467)  评论(0编辑  收藏  举报