hdu 3917 Road constructions

題意:有m個公司,k條路,每個公司修條路都要向政府交一定的稅,同時政府也要向該公司付修路費,還有如果a和b公司是合作關

系,那麼你選了a修路,就必須要選b。(如果a負責修A-B,b負責修B-C,那麼就定義a和b爲合作的關係),問政府最大收益爲多少。
最大權閉合圖:
建立超級源點和每個公司相連,權值爲該公司修路的稅費。
每個公司和其合作的公司相連,權值爲inf。
每個公司和匯點相連,權值爲它對應的修路費。
跑一遍sap(),
稅費總和-sap()即爲答案。



#include<stdio.h>   
#include<string.h>   
#include<iostream>   
#define N 40010        
#define inf 100000000       
using namespace std;  
struct Edge{      
int u,v,c,next;      
}edge[N];      
int head[N],pre[N],cur[N],dis[N],gap[N],a[N],b[N],company[N],c[N],s[N];      
int n,m,e,num,k,sum,start,end; 
void add_adge(int u,int v,int c)      
{      
   edge[e].u=u; edge[e].v=v; edge[e].c=c;  edge[e].next=head[u]; head[u]=e++;      
   edge[e].u=v; edge[e].v=u; edge[e].c=0;  edge[e].next=head[v]; head[v]=e++;      
}     
int sap()    
{    
   int flow=0,aug=inf,u;    
   bool flag;    
   for(int i=0;i<=m+1;i++)    
   {    
      cur[i]=head[i];    
      gap[i]=dis[i]=0;    
   }    
   gap[0]=m+2;    
   u=pre[start]=start;    
  while(dis[start]<m+2)    
  {    
      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==end)    
               {    
                  flow+=aug;    
                  while(u!=start)    
                  {    
                     u=pre[u];    
                     edge[cur[u]].c-=aug;    
                     edge[cur[u]^1].c+=aug;    
                  }    
                  aug=inf;    
               }    
                break;    
           }    
    
     }    
     if(flag)  continue;    
     int mindis=m+2;    
     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()
{
    while(scanf("%d%d",&n,&m)&&n&&m)
    {
        int i,j;
        if(n>m) end=n+1;
        else end=m+1;
        e=sum=start=0;
        memset(head,-1,sizeof(head));
        memset(s,0,sizeof(s));
        for(i=1;i<=m;i++)
        {
            scanf("%d",&num);
            add_adge(0,i,num);
            sum+=num;
        }
        scanf("%d",&k);
        for(i=1;i<=k;i++)
        {
            scanf("%d%d%d%d",&a[i],&b[i],&company[i],&c[i]);
            s[company[i]]+=c[i];
        }
        for(i=1;i<=k;i++)
           for(j=1;j<=k;j++)
           {
                 if(a[j]==b[i]&&company[i]!=company[j])
                      add_adge(company[i],company[j],inf);
           }
        for(i=1;i<=m;i++)
        {
            add_adge(i,end,s[i]);
        }
        printf("%d\n",sum-sap());
    } 
    return 0;
} 
           
                   
            

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