sgu 194 被动散热器具有最大流量的上限和下限(最大流量模板dinic加上优化)

模板类型的题详细參考国家集训队论文:http://wenku.baidu.com/view/0f3b691c59eef8c75fbfb35c.html
參考博客:http://blog.csdn.net/qq172108805/article/details/7783010
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
#define  N  300
#define inf 0x3fffffff
struct node {
   int u,v,w,next;
}bian[N*N*3];
int fee[N],bb[N*N*2];
int head[N],yong,dis[N],work[N];
void init(){
yong=0;
memset(head,-1,sizeof(head));
}
void addbian(int u,int v,int w) {
bian[yong].u=u;
bian[yong].v=v;
bian[yong].w=w;
bian[yong].next=head[u];
head[u]=yong++;
}
void add(int u,int v,int w) {
addbian(u,v,w);
addbian(v,u,0);
}
int min(int a,int b)
{
	return a<b?a:b;
}
int bfs(int s,int t)
{
    memset(dis,-1,sizeof(dis));
    queue<int>q;
    q.push(s);
    dis[s]=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u];i!=-1;i=bian[i].next)
        {
            int v=bian[i].v;
            if(bian[i].w&&dis[v]==-1)
            {
                dis[v]=dis[u]+1;
                q.push(v);
                if(v==t)
                    return 1;
            }
        }
    }
    return 0;
}
int dfs(int  s,int limit,int t)
{
    if(s==t)return limit;
    for(int &i=work[s];i!=-1;i=bian[i].next)
    {
        int v=bian[i].v;
        if(bian[i].w&&dis[v]==dis[s]+1)
        {
            int tt=dfs(v,min(limit,bian[i].w),t);
            if(tt)
            {
                bian[i].w-=tt;
                bian[i^1].w+=tt;
                return tt;
            }
        }
    }
    return 0;
}
int dinic(int s,int t)
{
    int ans=0;
    while(bfs(s,t))
    {
        memcpy(work,head,sizeof(head));
        while(int tt=dfs(s,inf,t))
            ans+=tt;
    }
    return ans;
}
int main(){
  int n,m,i,a,b,c,d,suma,f,s,t,w[N];
   while(scanf("%d%d",&n,&m)!=EOF) {
      init();
    s=0;t=n+1;
      suma=0;
      memset(w,0,sizeof(w));
     for(i=0;i<m;i++) {
        scanf("%d%d%d%d",&a,&b,&c,&d);
        add(a,b,d-c);
        w[b]+=c;
        w[a]-=c;
        bb[i]=d;
      }
      for(i=1;i<=n;i++) {
        if(w[i]>0) {
                suma+=w[i];
        add(s,i,w[i]);
        }
        if(w[i]<0)add(i,t,-w[i]);
      }
      f=dinic(s,t);
      if(f==suma) {
        printf("YES
");
        for(i=0;i<m;i++)
            printf("%d
",bb[i]-bian[i*2].w);
      }
      else
        printf("NO
");
   }
return 0;
}

版权声明:本文博主原创文章,博客,未经同意不得转载。

原文地址:https://www.cnblogs.com/mengfanrong/p/4869186.html