LightOJ 1154 - Penguins 最大流

题目：http://www.lightoj.com/volume_showproblem.php?problem=1154

题意：有n块浮冰，上面有一定数量的企鹅，企鹅每次跳跃时，会产生冲击力损坏起跳的浮冰，不损坏落下的浮冰，现在告诉你每块浮冰还能承受几次跳跃，问哪些浮冰能够让所有的企鹅聚齐。输入格式：首先一个t，代表测试数据组数，然后n d,代表浮冰块数和企鹅跳跃的最大距离，然后n行，每行 x y a b，分别代表浮冰坐标、浮冰上企鹅数、能承受冲击次数。

思路：首先预处理浮冰间的距离，因为要找到所有的满足条件的浮冰，那么我们枚举每个浮冰为汇点，从0向每个浮冰连边，容量为浮冰上企鹅数，对于每个浮冰，拆点连边，容量为能承受冲击的次数，限制经过的企鹅数。然后浮冰之间距离小于等于给定距离的连边，此时最大流的意义就是有几个企鹅能够到达汇点，如果等于企鹅总数，那就是满足条件的。还有浮冰编号是从0开始的

总结：拆点连边时一定要小心，不要连错，容量不要搞错！！！

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;

const int N = 210;
const int INF = 0x3f3f3f3f;
struct edge
{
    int to, cap, next;
}g[N*N*2];
int head[N], iter[N], level[N];
int n, cnt, _case = 0;
void add_edge(int v, int u, int cap)
{
    g[cnt].to = u, g[cnt].cap = cap, g[cnt].next = head[v], head[v] = cnt++;
    g[cnt].to = v, g[cnt].cap = 0, g[cnt].next = head[u], head[u] = cnt++;
}
bool bfs(int s, int t)
{
    memset(level, -1, sizeof level);
    level[s] = 0;
    queue<int> que;
    que.push(s);
    while(! que.empty())
    {
        int v = que.front(); que.pop();
        for(int i = head[v]; i != -1; i = g[i].next)
        {
            int u = g[i].to;
            if(g[i].cap > 0 && level[u] < 0)
            {
                level[u] = level[v] + 1;
                que.push(u);
            }
        }
    }
    return level[t] == -1;
}
int dfs(int v, int t, int f)
{
    if(v == t) return f;
    for(int &i = iter[v]; i != -1; i = g[i].next)
    {
        int u = g[i].to;
        if(g[i].cap > 0 && level[v] < level[u])
        {
            int d = dfs(u, t, min(g[i].cap, f));
            if(d > 0)
            {
                g[i].cap -= d, g[i^1].cap += d;
                return d;
            }
        }
    }
    return 0;
}
int dinic(int s, int t)
{
    int flow = 0, f;
    while(true)
    {
        if(bfs(s, t)) return flow;
        memcpy(iter, head, sizeof head);
        while(f = dfs(s, t, INF),f > 0)
            flow += f;
    }
}
int main()
{
    int t;
    int x[N], y[N], num[N], cut[N];
    double len, dis[N][N];
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%lf", &n, &len);
        int sum = 0;
        for(int i = 1; i <= n; i++)
        {
            scanf("%d%d%d%d", x + i, y + i, num + i, cut + i);
            sum += num[i];
        }
        for(int i = 1; i <= n; i++) //预处理浮冰间的距离
            for(int j = i + 1; j <= n; j++)
                dis[i][j] = dis[j][i] = sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]));
        bool flag = false;
        printf("Case %d: ", ++_case);
        for(int i = 1; i <= n; i++) //枚举所有浮冰为汇点
        {
            cnt = 0;
            memset(head, -1, sizeof head);
            for(int j = 1; j <= n; j++)
                add_edge(0, j, num[j]);
            for(int j = 1; j <= n; j++) //拆点
                add_edge(j, n + j, cut[j]);
            for(int j = 1; j <= n; j++)
                for(int k = j + 1; k <= n; k++)
                    if(dis[j][k] <= len)
                    {
                        add_edge(j + n, k, cut[j]); //从j到k，因为拆点的限制，最多能通过cut[j]的流量，设为INF也可以，不容易搞错
                        add_edge(k + n, j, cut[k]); //从k到j，最多通过cut[k]的流量
                    }
            if(dinic(0, i) == sum)
            {
                if(flag) printf(" %d", i - 1);
                else printf("%d", i - 1), flag = true;
            }
        }
        if(! flag) printf("-1");
        printf("\n");
    }
    return 0;
}