sgu 176 Flow construction 上下界最小流

来源：互联网发布：大理寺知世壁纸编辑：程序博客网时间：2024/05/22 03:52
对于上下限最大最小流可以参看本博客网络流里的一篇文章。
#include<stdio.h>#include<iostream>#include<algorithm>#include<vector>#include<string.h>#include<queue>#define LL int#define inf 100000000#define maxn 150using namespace std;struct pi{    int to;    int cost;    int rev;    int id;}pp;vector<pi >g[maxn];queue<int>q;int s,t;int line[maxn],leve[maxn];int min(int a,int b){    int p;    p=a;    if(b<a)        p=b;    return p;}void add(int a,int b,int cos,int id){    pp.to=b;    pp.cost=cos;    pp.rev=(int)g[b].size();    pp.id=id;    g[a].push_back(pp);    pp.to=a;    pp.cost=0;    pp.rev=(int)g[a].size()-1;    pp.id=-1;    g[b].push_back(pp);    return ;}void bfs(void){    q.push(s);    int p,f,i;    while(!q.empty())    {        p=q.front();        q.pop();        f=(int)g[p].size();        for(i=0;i<f;i++)        {            pi &e=g[p][i];            if(e.cost>0&&line[e.to]<0)            {                line[e.to]=line[p]+1;                q.push(e.to);            }        }    }    return ;}int dfs(int v,int f){    int p;    if(v==t)        return f;    for(int &i=leve[v];i<g[v].size();i++)    {        pi &e=g[v][i];        if(line[v]<line[e.to]&&e.cost>0)        {            p=dfs(e.to,min(f,e.cost));            if(p>0)            {                e.cost-=p;                g[e.to][e.rev].cost+=p;                return p;            }        }    }    return 0;}LL  dinic(){    int f;    LL flow=0;    while(1)    {        memset(line,-1,sizeof(line));        memset(leve,0,sizeof(leve));        line[s]=0;        bfs();        if(line[t]<0)            return flow;        f=dfs(s,inf);        if(f==0)            continue;        flow+=f;        while((f=dfs(s,inf))>0)        {            flow+=f;        }    }}struct ppi{    int from;    int to;    int a;    int b;}pp1[100005];char c[1005];int a[105];int b[100005];int main(){    int i,j,n,m,p,k,N=0,le,ri,mid;    while(scanf("%d%d",&n,&m)!=EOF){        memset(a,0,sizeof(a));        for(i=0;i<=n+1;i++) g[i].clear();        for(i=0;i<m;i++){            scanf("%d%d%d",&pp1[i].from,&pp1[i].to,&pp1[i].b);            scanf("%d",&p);            if(p==0){                pp1[i].a=0;            }            else pp1[i].a=pp1[i].b;            a[pp1[i].from]+=pp1[i].a;            a[pp1[i].to]-=pp1[i].a;        }        s=0;        t=n+1;        le=0;        ri=1000000000;        while(le<=ri){            mid=(le+ri)/2;            for(i=0;i<=t;i++) g[i].clear();            for(i=0;i<m;i++){                add(pp1[i].from,pp1[i].to,pp1[i].b-pp1[i].a,i);            }            for(i=1;i<=n;i++){                if(a[i]==0) continue;                if(a[i]>0) add(i,t,a[i],-1);                else add(s,i,-a[i],-1);            }            add(n,1,mid,-1);            dinic();            p=0;            k=g[s].size();            for(i=0;i<k;i++){                if(g[s][i].cost!=0){                    p=1;                    break;                }            }            if(p) le=mid+1;            else ri=mid-1;        }        if(le>=1000000000){            printf("Impossible\n");        }        else{            printf("%d\n",le);            for(i=0;i<=t;i++) g[i].clear();            for(i=0;i<m;i++){                add(pp1[i].from,pp1[i].to,pp1[i].b-pp1[i].a,i);            }            for(i=1;i<=n;i++){                if(a[i]==0) continue;                if(a[i]>0) add(i,t,a[i],-1);                else add(s,i,-a[i],-1);            }            add(n,1,le,-1);            dinic();            for(i=1;i<=n;i++){                k=g[i].size();                for(j=0;j<k;j++){                    if(g[i][j].id!=-1){                        b[g[i][j].id]=pp1[g[i][j].id].b-g[i][j].cost;                    }                }            }            for(i=0;i<m;i++){                if(i==0) printf("%d",b[i]);                else printf(" %d",b[i]);            }            printf("\n");        }    }}
0 0