程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> POJ 2391Ombrophobic Bovines（二分+最短路+網絡流之最大流）

POJ 2391Ombrophobic Bovines（二分+最短路+網絡流之最大流）

編輯：C++入門知識

題目地址：http://poj.org/problem?id=2391

這個題WA了一晚上，原因是數組開小了，然後又TLE了一天，原因是數組改的過大了。。。。不多說什麼了。。。

思路不難，建圖也不難，二分時間，然後把每個田地之間的最短距離用floyd最短路求出來。然後建立一個源點與匯點，將田地拆分成兩個點，在距離之內的進行連邊，要單向連邊。然後將源點與田地相連，權值為每個田地的牛的數目，再把另一邊的田地與匯點相連，權值為每個田地最大可避雨的牛的數目。拆開的田地之間權值可以為無窮大。

代碼如下：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

using namespace std;
const long long maxint1=1e16;
const int maxint=1e9;
long long dp[500][500],  sum, n;
int num[500], d[500], pre[10000], cur[500], q[500],head[100000], s, t, nv, cnt;
struct node
{
    int u, v, cap, next;
} edge[10000000];
struct filed
{
    int x, y;
} ff[300];
void add(int u, int v, int cap)
{
    edge[cnt].v=v;
    edge[cnt].cap=cap;
    edge[cnt].next=head[u];
    head[u]=cnt++;

    edge[cnt].v=u;
    edge[cnt].cap=0;
    edge[cnt].next=head[v];
    head[v]=cnt++;
}
void floyd()
{
    int i, j, k;
    for(k=1; k<=n; k++)
    {
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                if(dp[i][j]>dp[i][k]+dp[k][j])
                {
                    dp[i][j]=dp[i][k]+dp[k][j];
                }
            }
        }
    }
}
void bfs()
{
    memset(num,0,sizeof(num));
    memset(d,-1,sizeof(d));
    int f1=0, f2=0, i;
    d[t]=0;
    num[0]=1;
    q[f1++]=t;
    while(f1>=f2)
    {
        int u=q[f2++];
        for(i=head[u]; i!=-1; i=edge[i].next)
        {
            int v=edge[i].v;
            if(d[v]==-1)
            {
                d[v]=d[u]+1;
                num[d[v]]++;
                q[f1++]=v;
            }
        }
    }
}
int isap()
{
    memcpy(cur,head,sizeof(cur));
    int flow=0, i, u=pre[s]=s;
    bfs();
    while(d[s]edge[cur[i]].cap)
                {
                    f=edge[cur[i]].cap;
                    pos=i;
                }
            }
            for(i=s; i!=t; i=edge[cur[i]].v)
            {
                edge[cur[i]].cap-=f;
                edge[cur[i]^1].cap+=f;
            }
            flow+=f;
            if(flow>=sum)
                return flow;
            u=pos;
        }
        for(i=cur[u]; i!=-1; i=edge[i].next)
        {
            if(d[edge[i].v]+1==d[u]&&edge[i].cap)
            {
                break;
            }
        }
        if(i!=-1)
        {
            cur[u]=i;
            pre[edge[i].v]=u;
            u=edge[i].v;
        }
        else
        {
            if((--num[d[u]])==0) break;
            int mind=nv;
            for(i=head[u]; i!=-1; i=edge[i].next)
            {
                if(mind>d[edge[i].v]&&edge[i].cap)
                {
                    mind=d[edge[i].v];
                    cur[u]=i;
                }
            }
            d[u]=mind+1;
            num[d[u]]++;
            u=pre[u];
        }
    }
    return flow;
}
int main()
{
    long long i, a, b, j;
    long long m, c;
    //freopen("1.txt","r",stdin);
    //freopen("2.txt","w",stdout);
    scanf("%lld%lld",&n,&m);
    sum=0;
    for(i=0; i<=n; i++)
    {
        for(j=0; j<=n; j++)
        {
            dp[i][j]=maxint1;
            if(i==j)
            {
                dp[i][j]=0;
            }
        }
    }
    for(i=1; i<=n; i++)
    {
        scanf("%d%d",&ff[i].x,&ff[i].y);
        sum+=ff[i].x;
    }
    while(m--)
    {
        scanf("%lld%lld%lld",&a,&b,&c);
        if(dp[a][b]>c)
        {
            dp[a][b]=c;
            dp[b][a]=c;
        }
    }
    floyd();
    int x;
    long long high=200000000000, mid, low=1, ans=-1;
    while(low<=high)
    {
        mid=(high+low)/2;
        s=0;
        t=2*n+1;
        nv=t+1;
        memset(head,-1,sizeof(head));
        cnt=0;
        for(i=1; i<=n; i++)
        {
            add(s,i,ff[i].x);
            add(i+n,t,ff[i].y);
        }
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                if(dp[i][j]<=mid)
                {
                    add(i,j+n,maxint);
                }
            }
        }
        x=isap();
        if(x>=sum)
        {
            ans=mid;
            high=mid-1;
        }
        else
        {
            low=mid+1;
        }
    }
    printf("%lld\n",ans);
    return 0;
}