PKU 3281 Dining

原创于 2017-09-29 22:06:24 发布 · 183 阅读

0 ·

CC 4.0 BY-SA版权

PKU 专栏收录该内容

4 篇文章

订阅专栏

本文详细解析了POJ 3281题目中的Dinic算法实现，重点介绍了如何进行图的构建，并通过具体代码展示了从输入到求解的全过程。

题目链接：

http://poj.org/problem?id=3281

题解：

Dinic模板关键是怎么建图。但是自己比较蛇皮的只建了一只牛的图，到现在还没怎么明白为什么要建两只牛。

代码：

#include <cmath>
#include <map>
#include <vector>
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define inf 0x7fffffff
#define met(a,b) memset(a,b,sizeof(a))
const int maxn =1000+10;
int mp[maxn][maxn];
int dis[maxn];
int s,t;

int bfs()
{
    queue<int>q;
    met(dis,0);
    dis[s]=1;
    q.push(s);
    while(!q.empty())
    {
        int d=q.front();
        q.pop();
        for(int i=s;i<=t;i++)
        {
            if(!dis[i]&&mp[d][i])
            {
                dis[i]=dis[d]+1;
                q.push(i);
            }
        }
    }
    return dis[t];
}

int dfs(int pos,int flow)
{
    if(pos==t)
        return flow;
    int temp=flow;
    for(int i=0;i<=t&&temp;i++)
    {
        if(dis[i]==dis[pos]+1&&mp[pos][i])
        {
            int a=dfs(i,min(temp,mp[pos][i]));
            mp[pos][i]-=a;
            mp[i][pos]+=a;
            temp-=a;
        }
    }
    return flow-temp;
}

int Dinic()
{
    int ans=0,temp=0;
    while(bfs())
    {
        while(temp=dfs(s,inf))
            ans+=temp;
    }
    return ans;
}

int main()
{
    int N,F,D;
    scanf("%d%d%d",&N,&F,&D);
    s=0,t=2*N+F+D+1;
    for(int i=1;i<=N;i++)
    {
        int f,d;
        scanf("%d%d",&f,&d);
        for(int j=0;j<f;j++)
        {
            int x;
            scanf("%d",&x);
            x+=2*N;
            mp[x][i]=1;
            mp[s][x]=1;
        }
        for(int j=0;j<d;j++)
        {
            int x;
            scanf("%d",&x);
            x+=2*N+F;
            mp[i+N][x]=1;
            mp[x][t]=1;
        }
    }
    for(int i=1;i<=N;i++)
        mp[i][i+N]=1;
    printf("%d\n",Dinic());
}