POJ 2455Secret Milking Machine（二分+网络流之最大流）

标签：c语言   算法   编程   网络流   poj   

题目地址：POJ2455

手残真浪费时间啊。。又拖到了今天才找出了错误。。每晚两道题不知不觉又变回了每晚一道题、。。sad。。

第一次在isap中忘记调用bfs，第二次则是遍历的时候居然是从1开始遍历的。。。sad。。。

这题思路倒是很简单，就是有一个比较坑的地方，就是这里的重边要当两条边来用，以前受最短路什么的影响，直接把慢的删了，很明显不对。。。这里的两条重边都是可以走的。

建图思路是，将1当作源点，将n当作汇点。右边的地方就连边，注意是无向边。最后判断最大流是否等于道路条数。二分找最小值。

代码如下：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <algorithm>

using namespace std;
const int INF=1e9;
int head[500], s, t, nv, cnt, sum;
int num[500], d[500], pre[500], cur[500], q[500];
struct node
{
    int u, v, cap, next;
}edge[5000000];
struct N
{
    int u, v, w;
}bian[5000000];
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=cap;
    edge[cnt].next=head[v];
    head[v]=cnt++;
}
void bfs()
{
    memset(num,0,sizeof(num));
    memset(d,-1,sizeof(d));
    int f1=0, f2=0, i;
    q[f1++]=t;
    d[t]=0;
    num[0]=1;
    while(f1>=f2)
    {
        int u=q[f2++];
        for(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[f1++]=v;
            }
        }
    }
}
int isap()
{
    memcpy(cur,head,sizeof(cur));
    int flow=0, u=pre[s]=s, i;
    bfs();
    while(d[s]<nv)
    {
        if(u==t)
        {
            int f=INF, pos;
            for(i=s;i!=t;i=edge[cur[i]].v)
            {
                if(f>edge[cur[i]].cap)
                {
                    f=edge[cur[i]].cap;
                    pos=i;
                }
            }
            for(i=s;i!=t;i=edge[cur[i]].v)
            {
                edge[cur[i]].cap-=f;
                edge[cur[i]^1].cap+=f;
            }
            flow+=f;
            if(flow>=sum)
            {
                return flow;
            }
            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];
        }
    }
    return flow;
}
int main()
{
    int n, p, a, b, c, i, j, top;
    while(scanf("%d%d%d",&n,&p,&sum)!=EOF)
    {
        top=0;
        while(p--)
        {
            scanf("%d%d%d",&a,&b,&c);
            bian[top].u=a;
            bian[top].v=b;
            bian[top++].w=c;
        }
        int high=1000001, mid, low=1, ans, x;
        while(low<=high)
        {
            mid=(high+low)/2;
            s=1;
            t=n;
            nv=t+1;
            memset(head,-1,sizeof(head));
            cnt=0;
            for(i=0;i<top;i++)
            {
                if(bian[i].w<=mid)
                {
                    add(bian[i].u,bian[i].v,1);
                }
            }
            x=isap();
            //printf("%d %d %d\n",x, mid,top);
            if(x>=sum)
            {
                ans=mid;
                high=mid-1;
            }
            else
            {
                low=mid+1;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

POJ 2455Secret Milking Machine（二分+网络流之最大流）,布布扣,bubuko.com

POJ 2455Secret Milking Machine（二分+网络流之最大流）

标签：c语言   算法   编程   网络流   poj   

原文地址：http://blog.csdn.net/scf0920/article/details/36180387