poj 3281

终于在有模版 有算法书的情况下 倒腾了半天 a了人生中第一道最大流


题意:

有n头牛,f种食物, d种饮料,每头牛都有自己喜欢的饮料和食物,问最多能让几头牛同时得到喜欢的食物和饮料.


关键是建图 增加一个超级汇源 s  一个超级汇点t

然后 s 与食物连边 饮料与t 连边  为了防止一头牛不会被多组食物分配 所以将牛拆开


然后就是  s  -  食物 - 牛 -  牛 - 饮料 - t


然后就是套最大流模版 


#include <iostream>
#include <queue>
#include <cstring>
#include <cstdio>
#include <vector>
#include <string>
#include <iterator>
#include <cmath>
#include <deque>
#include <stack>
#include <cctype>
#include <iomanip>
using namespace std;

typedef long long ll;
typedef long double ld;

const int N = 1100;
const int INF = 0xfffffff;

#define INFL 0x7fffffffffffffffLL
#define met(a, b) memset(a, b, sizeof(a))

struct edge{int from, to, cap, flow;};
vector <edge> edges; //边表
vector <int> G[N]; //邻接表 G[i][j] 结点i的第j条边在e数组中的序号
bool vis[N]; // bfs使用
int d[N]; // 从起点到i的距离
int cur[N]; //当前弧下标
bool liked[N][N], likef[N][N];

void addedge (int from, int to, int cap)
{
    edges.push_back((edge){from, to, cap, 0});
    edges.push_back((edge){to, from, 0, 0});
    int tm = edges.size();
    G[from].push_back(tm-2);
    G[to].push_back(tm-1);
}
bool bfs (int s, int t)
{
    met (vis, false);
    queue <int> que;
    que.push (s);
    d[s] = 0;
    vis[s] = true;
    while (que.size())
    {
        int x = que.front(); que.pop();
        for (int i=0; i<G[x].size(); i++)
        {
            edge &e = edges[G[x][i]];
            if (!vis[e.to] && e.cap > e.flow)
            {
                vis[e.to] = 1;
                d[e.to] = d[x] + 1;
                que.push(e.to);
            }
        }
    }
    return vis[t];
}

int dfs (int x, int t, int a)
{
    if (x == t || a == 0) return a;
    int flow = 0, f;
    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, t, 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 s, int t)
{
    int flow = 0;
    while (bfs(s, t))
    {
        met (cur, 0);
        flow += dfs(s, t, INF);
    }
    return flow;
}

void solve (int n1, int f1, int d1)
{
    // 0 ~ n1-1 食物一侧的牛
    // n1 ~ 2*n1-1:饮料一侧的牛
    // 2*n1 ~ 2*n1+f-1 食物
    //2*n1+f ~ 2*n1+f+d-1 饮料
    int s = n1 * 2 + f1 + d1, t = s + 1;
    for (int i=0; i<f1; i++) //s 与 食物
        addedge(s, n1*2+i, 1);

    for (int i=0; i<d1; i++) //饮料 和 t
        addedge(n1*2+f1+i, t, 1);

    for (int i=0; i<n1; i++) //食物和左侧的牛
    {
        addedge(i, n1+i, 1);
        for (int j=0; j<f1; j++)
        {
            if (likef[i][j])
                addedge(n1*2+j, i, 1);
        }
        for (int j=0; j<d1; j++)
        {
            if (liked[i][j])
                addedge(n1+i, n1*2+f1+j, 1);
        }
    }
    cout << maxflow (s, t) << endl;
}

int main ()
{
    int n1, d1, f1;
    while (cin >> n1 >> f1 >> d1)
    {
        int x, y, tx, ty;
        met (liked, false);
        met (likef, false);
        met (G, 0);
        edges.clear();

        for (int i=0; i<n1; i++)
        {
            cin >> x >> y;
            while (x--)
            {
                cin >> tx;
                likef[i][tx-1] = true;
            }
            while (y--)
            {
                cin >> ty;
                liked[i][ty-1] = true;
            }
        }
        solve (n1, f1, d1);
    }
    return 0;
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值