其他分享
首页 > 其他分享> > SCUT - 205 - 饲养牛 - 最大流

SCUT - 205 - 饲养牛 - 最大流

作者:互联网

https://scut.online/p/205
连着做所以一开始就觉得是网络流。
这种至多分配几次的很有网络流的特征。

一开始想从食物和饮料流向牛,但是怎么搞都不对。
其实可以从s流向食物,食物流向牛,牛流向饮料,饮料流向t。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;


namespace Dinic_Maxflow {

    /* 备注:
    1.检查MAXN与MAXM,注意预留反向边和额外边的位置
    2.每次建图的第一次add_edge()前必须先init()
    3.不传入第三参数的dinic(s,t)需保证t是最后的结点
     */

    const int INF=0x3f3f3f3f;

    const int MAXN=2000;
    const int MAXM=200000;
    //注意网络流要预留反向边

    int tol;
    int head[MAXN+5];

    struct Edge {
        int to,next,cap,flow;
    } edge[MAXM+5];


    void init() {
        //.//用来触发编译错误
        tol=2;
        memset(head,-1,sizeof(head));
    }

    //为了方便使用二分图匹配,默认容量为1,默认反向容量为0
    void add_edge(int u,int v,int w=1,int rw=0) {
        edge[tol].to=v;
        edge[tol].cap=w;
        edge[tol].flow=0;
        edge[tol].next=head[u];
        head[u]=tol++;
        edge[tol].to=u;
        edge[tol].cap=rw;
        edge[tol].flow=0;
        edge[tol].next=head[v];
        head[v]=tol++;
    }

    int Q[MAXN+5];
    int dep[MAXN+5],cur[MAXN+5],sta[MAXN+5];

    bool bfs(int s,int t) {
        //front 与 back 的缩写
        int fnt=0,bak=0;
        memset(dep,-1,sizeof(dep[0])*(t+1));
        dep[s]=0;
        Q[bak++]=s;
        while(fnt<bak) {
            int u=Q[fnt++];
            for(int i=head[u]; i!=-1; i=edge[i].next) {
                int v=edge[i].to;
                if(dep[v]==-1&&edge[i].cap>edge[i].flow) {
                    dep[v]=dep[u]+1;
                    if(v==t)
                        return true;
                    Q[bak++]=v;
                }
            }
        }
        return false;
    }

    int dinic(int s,int t) {
        int maxflow=0;
        while(bfs(s,t)) {
            for(int i=0; i<=t; i++)
                cur[i]=head[i];
            int u=s,tail=0;
            while(cur[s]!=-1) {
                if(u==t) {
                    int tp=INF;
                    for(int i=tail-1; i>=0; i--) {
                        tp=min(tp,edge[sta[i]].cap-edge[sta[i]].flow);
                    }
                    maxflow+=tp;
                    for(int i=tail-1; i>=0; i--) {
                        edge[sta[i]].flow+=tp;
                        edge[sta[i]^1].flow-=tp;
                        if(edge[sta[i]].cap-edge[sta[i]].flow==0)
                            tail=i;
                    }
                    u=edge[sta[tail]^1].to;
                } else if(cur[u]!=-1&&edge[cur[u]].cap>edge[cur[u]].flow
                          &&dep[u]+1==dep[edge[cur[u]].to]) {
                    sta[tail++]=cur[u];
                    u=edge[cur[u]].to;
                } else {
                    while(u!=s&&cur[u]==-1) {
                        u=edge[sta[--tail]^1].to;
                    }
                    cur[u]=edge[cur[u]].next;
                }
            }
        }
        return maxflow;
    }

    /* 备注:
    1.检查MAXN与MAXM,注意预留反向边和额外边的位置
    2.每次建图的第一次addedge()前必须先init()
    3.不传入第三参数的dinic(s,t)需保证t是最后的结点
     */

}


using namespace Dinic_Maxflow;

int n,f,d;
inline int id(char type,int x) {
    int res=0;
    switch(type) {
        case 'n':
            return x;
        case 'f':
            return n+x;
        case 'd':
            return n+f+x;
        default:
            exit(-1);
    }
}

int main() {
#ifdef Yinku
    freopen("Yinku.in","r",stdin);
#endif // Yinku
    while(~scanf("%d%d%d",&n,&f,&d)) {
        init();
        int s=0,t=n+f+d+1;
        for(int i=1;i<=f;i++){
            add_edge(s,id('f',i),1);
        }
        for(int i=1;i<=d;i++){
            add_edge(id('d',i),t,1);
        }
        for(int i=1; i<=n; i++) {
            int fi,di;
            scanf("%d%d",&fi,&di);
            int u=id('n',i);
            for(int j=1;j<=fi;j++){
                int v;
                scanf("%d",&v);
                v=id('f',v);
                add_edge(v,u,1);
            }
            for(int j=1;j<=di;j++){
                int v;
                scanf("%d",&v);
                v=id('d',v);
                add_edge(u,v,1);
            }
        }

        printf("%d\n",dinic(s,t));
    }
}

标签:SCUT,205,cur,int,flow,饲养,edge,tol,sta
来源: https://www.cnblogs.com/Yinku/p/11024942.html