2018-acmicpc-南京 I-Magic Potion

案例1最大流建图

1 - n表示英雄，n+1 - m+n 表示怪兽， 0表示起点（简称S点）， n+m+1表示终点（E点），n+m+2表示K瓶药水所代表的点（K点）。

说明

• 起点 -> K点的边，流量为k，其他边流量为1：象征最多使用k瓶药水。
• K点 -> 英雄点，流量为1：每个英雄最多使用一瓶药水。
• 怪兽点 -> E点，流量为1：每个怪兽最多只能被杀一次。

代码

#include <bits/stdc++.h>

#define Debug 0
#define MAXN 1056
#define MAXM 2600
#define MOD 1000000007
#define INF 0x3f3f3f3f
#define PI 3.1415926535
#define pb push_back
#define SYNC ios::sync_with_stdio(false);
//#define FILE_IN "I:\\acm-workspace\\CLion-Project\\in.txt"
typedef long long ll;
typedef std::pair<int, int> Pair;

using namespace std;

int n, m, k;
int S, E, K;
int G[MAXN][MAXN];
int dep[MAXN];

void add_edge (int u, int v, int f)
{
 // 依题意，单向边，双向会WA
//    G[u][v] = G[v][u] = f; 
    G[u][v] = f;
}

bool bfs (int s, int e)
{
    memset(dep, -1, sizeof(dep));
    queue<int> que;
    dep[s] = 0;
    que.push(s);
    while (!que.empty())
    {
        int u = que.front();
        que.pop();
        for (int i = 0; i < MAXN; ++i)
        {
            int v = i;
            int flow = G[u][i];
            if (v != u && dep[v] == -1 && flow > 0)
            {
                dep[v] = dep[u] + 1;
                que.push(v);
            }

        }
    }
    return dep[e] != -1;
}

int dfs (int u, int flow)
{
    if (u == E || flow == 0)
    {
        return flow;
    }
    int max_flow = 0, find_flow;
    for (int i = 0; i < MAXN; ++i)
    {
        int v = i;
        int f = G[u][i];
        if (v != u && f > 0 && dep[v] == dep[u] + 1)
        {
            find_flow = dfs(v, min(flow - max_flow, f));
            if (find_flow > 0)
            {
                G[u][v] -= find_flow;
                G[v][u] += find_flow;
                max_flow += find_flow;
                if (max_flow == flow)
                    return flow;
            }
        }
    }
    if (max_flow == 0)
        dep[u] = -1;
    return max_flow;
}

int dinic ()
{
    int max_flow = 0;
    while (bfs(S, E))
    {
        max_flow += dfs(S, INF);
    }
    return max_flow;
}

int main ()
{

#ifdef FILE_IN
    freopen(FILE_IN, "r", stdin);
#endif
//    SYNC

    cin >> n >> m >> k;
    S = 0, E = n + m + 1, K = n + m + 2;
    memset(G, 0, sizeof(G));
    add_edge(S, K, k);
    for (int i = 1; i <= n; ++i)
    {
        add_edge(S, i, 1);
        add_edge(K, i, 1);
        int t;
        cin >> t;
        for (int j = 0; j < t; ++j)
        {
            int v;
            cin >> v;
            add_edge(i, n + v, 1);
        }

    }
    for (int i = 1; i <= m; ++i)
    {
        add_edge(i + n, E, 1);
    }

    int ans = dinic();

    cout << ans << endl;

#ifdef FILE_IN
    fclose(stdin);
#endif

    return 0;
}