hdu 3987 Harry Potter and the Forbidden Forest

最小割:
我用的比較傳統的方法,因爲最小割=最大流,所以可以先用sap預處理一下圖,找出滿流的邊,在把這下滿流的邊的權值更改爲1,其餘的變成inf,再跑一遍sap即可得到答案。



#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
const int N=2010;
const int M=550000;
const int inf=(1<<28);
int head[N];
int cnt,n,s,T,t,e,cas,m,u,v,c,d;
struct Edge
{
    int v,c,next;
}edge[M];

void add_edge(int u,int v,int c)    
{    
     edge[e].v=v;  edge[e].c=c;   edge[e].next=head[u];   head[u]=e++;      
     edge[e].v=u;  edge[e].c=0;   edge[e].next=head[v];   head[v]=e++;    
}   
   
int sap()
{
    int pre[N],cur[N],dis[N],gap[N];
    int flow=0,aug=inf,u;
    bool flag;
    for(int i=0; i<n; i++)
    {
        cur[i]=head[i];
        gap[i]=dis[i]=0;
    }
    gap[s]=n;//dis爲0的有n個 
    u=pre[s]=s;
    while(dis[s]<n)
    {
        flag=0;
        for(int &j=cur[u]; j!=-1; j=edge[j].next)
        {
            int v=edge[j].v;
            if(edge[j].c>0&&dis[u]==dis[v]+1)
            {
                flag=1;
                if(edge[j].c<aug) aug=edge[j].c;
                pre[v]=u;
                u=v;
                if(u==t)
                {
                    flow+=aug;
                    while(u!=s)
                    {
                        u=pre[u];
                        edge[cur[u]].c-=aug;
                        edge[cur[u]^1].c+=aug;
                    }
                    aug=inf;
                }
                break;
            }
        }
        if(flag) continue;
        int mindis=n;
        for(int j=head[u]; j!=-1; j=edge[j].next)
        {
            int v=edge[j].v;
            if(edge[j].c>0&&dis[v]<mindis)
            {
                mindis=dis[v];
                cur[u]=j;
            }
        }
        if((--gap[dis[u]])==0)
            break;
        gap[dis[u]=mindis+1]++;
        u=pre[u];
    }
    return flow;
}

int main()
{
    cas=0;
    scanf("%d",&T);
    while(T--)
    {
       cas++;
       scanf("%d%d",&n,&m);
       e=0;s=0;t=n-1;
       memset(head,-1,sizeof(head));
       for(int i=0;i<m;i++)
       {
          scanf("%d%d%d%d",&u,&v,&c,&d);
          if(d) add_edge(v,u,c);
          add_edge(u,v,c);
       }
       int pp = sap();
       for(int i=0;i<e;i++)
       {
            if(i&1) edge[i].c=0;
            else {
                 if(edge[i].c==0){
                                  edge[i].c=1;
                                  edge[i^1].c=0;
                }
                else {
                      edge[i].c=inf;
                      edge[i^1].c=0;
                }
            }
       }
       int answer=sap();
       printf("Case %d: %d\n",cas,answer);
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章