POJ 3281 Dining(最大流)

POJ 3281 Dining(最大流)

题目链接：poj3281

       此题的建模方法比较有开创性。以往一般都是左边一个点集表示供应并与源相连，右边一个点集表示需求并与汇相连。现在不同了，供应有两种资源，需求仍只有一个群体，怎么办？其实只要仔细思考一下最大流的建模原理，此题的构图也不是那么难想。最大流的正确性依赖于它的每一条 s-t 流都与一种实际方案一一对应。那么此题也需要用 s-t 流将一头牛和它喜欢的食物和饮料“串”起来，而食物和饮料之间没有直接的关系，自然就想到把牛放在中间，两边是食物和饮料，由 s, t 将它们串起来构成一种分配方案。至此建模的方法也就很明显了：每种食物 i 作为一个点并连边(s, i, 1)，每种饮料 j 作为一个点并连边(j, t, 1)，将每头牛 k拆成两个点 k’, k’’并连边(k’, k’’, 1), (i, k’, 1), (k’’, j, 1)，其中 i, j 均是牛 k 喜欢的食物或饮料。求一次最大流即为结果。

这边需要注意下的是结点的位置，看很多博客都是非常严谨的计算结点个数，我觉得最后结点个数反正也不会超过500，既然如此那何必要斤斤计较结点的安排。

因为这里的牛和食物和饮料都最多为100，所以我设源点为0，汇点为450.

食物：第i个食物的结点为i

食物牛：第i个食物牛的结点为i+110

饮料牛：第i个饮料牛的结点为i+220

饮料：第i个饮料的结点为i+330

如此不是方便很多。

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
#define INF 1e9
using namespace std;
const int maxn=600;

struct Edge
{
    Edge(){}
    Edge(int from,int to,int cap,int flow):from(from),to(to),cap(cap),flow(flow){}
    int from,to,cap,flow;
    //cap是容量  flow是流量
};

struct Dinic
{
    int n,m,s,t;            //结点数,边数(包括反向弧),源点与汇点编号
    vector<Edge> edges;     //边表 edges[e]和edges[e^1]互为反向弧
    vector<int> G[maxn];    //邻接表,G[i][j]表示结点i的第j条边在e数组中的序号
    bool vis[maxn];         //BFS使用,标记一个节点是否被遍历过
    int d[maxn];            //从起点到i点的距离
    int cur[maxn];          //当前弧下标

    void init(int n,int s,int t)
    {
        this->n=n,this->s=s,this->t=t;
        for(int i=0;i<=n;i++) G[i].clear();
        edges.clear();
    }

    void AddEdge(int from,int to,int cap)
    {
        edges.push_back( Edge(from,to,cap,0) );
        edges.push_back( Edge(to,from,0,0) );//这条是反向边
        m = edges.size();//总共边的数量
        G[from].push_back(m-2);//正向边在edges数组中的编号
        G[to].push_back(m-1);//反向边在edges数组中的编号
    }

    bool BFS()
    {
        memset(vis,0,sizeof(vis));
        queue<int> Q;//用来保存节点编号的
        Q.push(s);
        d[s]=0;
        vis[s]=true;
        while(!Q.empty())
        {
            int x=Q.front(); Q.pop();
            for(int i=0; i<G[x].size(); i++)//x所连的边
            {
                Edge& e=edges[G[x][i]];//取出所连的那条边
                if(!vis[e.to] && e.cap>e.flow)//如果还没有经过  且流量小于容量 还可以再通
                {
                    vis[e.to]=true;
                    d[e.to] = d[x]+1;
                    Q.push(e.to);
                }
            }
        }
        return vis[t];//表示能否到大t点  不能的话返回false
    }

    int DFS(int x,int a)
    {
        if(x==t || a==0)return a;//到大汇点或者流量为0时返回
        int flow=0,f;//flow用来记录从x到t的最小残量
        for(int& i=cur[x]; i<G[x].size(); i++)
        {
            Edge& e=edges[G[x][i]];
            if(d[x]+1==d[e.to] && (f=DFS( e.to,min(a,e.cap-e.flow) ) )>0 )
			//判断是否是可行边 并对其进行深搜 是否有结果
            {
                e.flow +=f;
                edges[G[x][i]^1].flow -=f;//与其对应的反向边流量减少  以便于后悔恢复
                flow += f;
                a -= f;
                if(a==0) break;
            }
        }
        return flow;
    }

    int Maxflow()
    {
        int flow=0;
        while(BFS())
        {
            memset(cur,0,sizeof(cur));
            flow += DFS(s,INF);
        }
        return flow;
    }
}DC;


int main()
{
        int n,f,d;
        while(~scanf("%d%d%d",&n,&f,&d))
        {
        	DC.init(451,0,450);
        	for(int i=1;i<=f;i++)DC.AddEdge(0,i,1);//放外边处理  放下面边输入边处理的话会有重边
        	for(int i=1;i<=d;i++)DC.AddEdge(i+330,450,1);
			for(int i=1;i<=n;i++)
			{
				int a,b,c;
				scanf("%d%d",&a,&b);
				for(int j=1;j<=a;j++)//食物
				{
					scanf("%d",&c);
					DC.AddEdge(c,i+110,1);//食物到牛

				}
				for(int j=1;j<=b;j++)//饮料
				{
					scanf("%d",&c);
					DC.AddEdge(i+220,c+330,1);//牛到饮料
				}

				DC.AddEdge(i+110,i+220,1);//食物牛到饮料牛
			}
			printf("%d\n",DC.Maxflow());
        }
    return 0;
}