ZOJ 3792 Romantic Value(网络流之最小割)(找割边)

题目地址:ZOJ 3792

最小割做的太少。。这题很明显是找割边。找割边就是判断正向弧是否是0.如果跑完一次最小割后正向弧流量为0的话,那就说明这个边为一条割边。但是找到了割边后再怎么办呢。。中午睡觉的时候突然来了灵感。。再利用这些割边求一次最大流不就行了。。把割边的流量都设为1,其他的都为正无穷。那最后的流量就是最少需要的割边了。然后计算就可以了。

这次又把上限值直接设为sink+1了。。。导致WA了12发。。。。。sad。。。以后得注意。。。

代码如下:

#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 head[100], source, sink, nv, cnt;
int cur[100], num[100], d[100], pre[100];
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(num,0,sizeof(num));
    memset(d,-1,sizeof(d));
    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);
            }
        }
    }
}
int isap()
{
    memcpy(cur,head,sizeof(cur));
    int flow=0, u=pre[source]=source, i;
    bfs();
    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];
        }
    }
    return flow;
}
int main()
{
    int n, m, sum, u, v, w, t, k, i, j;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d%d",&n,&m,&source,&sink);
        memset(head,-1,sizeof(head));
        cnt=0;
        sum=0;
        nv=n+1;
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
            sum+=w;
        }
        int x=isap();
        if(x==0)
        {
            printf("Inf\n");
            continue ;
        }
        for(i=0; i<cnt; i+=2)
        {
            if(edge[i].cap==0)
            {
                edge[i].cap=1;
                edge[i^1].cap=0;
            }
            else
            {
                edge[i].cap=INF;
                edge[i^1].cap=0;
            }
        }
        int tot=isap();
        double ans=(sum-x)*1.0/tot;
        printf("%.2lf\n",ans);
    }
    return 0;
}


你可能感兴趣的:(编程,算法,C语言,ACM)