求SGU 194即ZOJ 2314题AC代码

求SGU 194即ZOJ 2314题AC代码

最近在搞网络流，刚研究完几个求最大流的基本算法，今天研究了一下有上下界的可行流问题，参考了大牛ADN.cn的图论总结，算法倒不是很难，代码也很快就写好了，只是不知道原因，就是不能AC，郁闷我一晚上。。。在SGU上 只能过9组数据，也不知道被什么BT的数据给卡住了，希望大牛们能给我点指点。 我用的方法是添加两个附加源汇，用dinic算法求最大流，如果源汇上的边满流，则说明有可行流，输出每条边的流量，否则输出NO.

下面是我WA掉的代码：

#include < iostream >
#include < cmath >
#include < algorithm >
#include < cstdio >
#include < cstring >
#include < cassert >
using   namespace  std;
#define  MAX 210
#define  MAXINT 999999999

struct  pipe
{
    
    __int64 a;
    __int64 b;
    __int64 l;
    __int64 r;
} mypipe[MAX * 1000 ];

__int64 n,m;
__int64 graph[MAX][MAX];
__int64 d[MAX];
__int64 ans = 0 ;

__int64 min(__int64 a,__int64 b)
{
    
    if(a<b)
        return a;
    else
        return b;
}

void  init()
{
    
    __int64 i;
    memset(graph,0,sizeof(graph));
    
    for(i=1;i<=m;i++)
    {
        scanf("%I64d%I64d%I64d%I64d",&mypipe[i].a,&mypipe[i].b,&mypipe[i].l,&mypipe[i].r);
        graph[mypipe[i].a][mypipe[i].b]+=mypipe[i].r-mypipe[i].l;
        graph[mypipe[i].a][n+2]+=mypipe[i].l;
        graph[n+1][mypipe[i].b]+=mypipe[i].l;
    }
}
bool  build(__int64 n)
{
    
    memset(d,0,sizeof(d));
    bool visit[MAX]={0};
    __int64 myqueue[MAX*MAX]={0};
    
    __int64 front=1,rear=1;
    myqueue[rear]=n-1;visit[myqueue[front]]=true;
    while(front<=rear) 
    {
        
        __int64 i;
        for(i=1;i<=n;i++)
        {
            if((!visit[i])&&graph[myqueue[front]][i]>0)
            {
                rear++;
                myqueue[rear]=i;
                d[i]=d[myqueue[front]]+1;
                visit[i]=true;
                
            }
        }
        
        front++;
    }
    if(visit[n]==true)
        return true;
    else
        return false;
}

__int64 dinic(__int64 v,__int64 sum,__int64 n)
{
    if(v==n)
        return sum;
    __int64 ret=0;
    __int64 t;
    __int64 i;
    for(i=1;i<=n;i++)
    {
        if(d[i]==d[v]+1&&graph[v][i]>0)
        { 
            t=dinic(i,min(graph[v][i],sum)-ret,n);
            graph[v][i]-=t; 
            graph[i][v]+=t;
            ret+=t;
        }
    } 
    return ret;
}


int  main()
{
    
    __int64 j;
    __int64 flag=0;
    
    
    while(scanf("%I64d%I64d",&n,&m)!=EOF)
    {

        
        
        init();
        flag=0;
        while(build(n+2))
        {
            dinic(n+1,MAXINT,n+2);
        }
        for(j=1;j<=n;j++)
        {
            
            if(graph[j][n+2]!=0||graph[n+1][j]!=0)
            {
                
                flag=1;
                break;
            }
        }
        if(flag==1)
        {
            printf("NO\n\n");
            continue;
        }
        else
        {
            printf("YES\n");
            for(j=1;j<=m;j++)
            {
                
                printf("%I64d\n",graph[mypipe[j].b][mypipe[j].a]+mypipe[j].l);
            }
            printf("\n");
        }
        
    }
    return 0;
}