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;}