LightOJ 1398 Winger Trial 最大流

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

题意：一个L * W的矩形场地，在第一象限，左下为（0,0），右上为（L,W），上面有n个机器人，给出了它们的坐标，当距离机器人的距离小于等于d时，机器人会拦截。求当一个人从矩形左侧穿过矩形走到右侧时，最少会被拦截几次

思路：刚开始想从左到右直接建图，结果实在无法定义流的意义，后来一想，此题求的是纵向上有几条机器人的攻击范围形成的隔断，应该上下建图，把下侧边界和上侧边界分别作为源点和汇点，对于和边界距离小于d的机器人，连边，容量为1，如果两个机器人之间的距离小于2d，从这两个机器人之间经过时，一定会被拦截一次，那么连边，容量为1，那么此时最大流的意义就是纵向上有几条机器人攻击范围形成的隔断，测样例也对，于是愉快的交了，结果哇了，在仔细一分析，会有如下情况：大于等于两条的隔断交于一点的话，那么从交点走过，只会被拦截一次，而我的写法就错了，于是对每个点拆点连边，容量为1，就可以解救这种情况。

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

const int N = 210;
const int INF = 0x3f3f3f3f;
struct edge
{
    int to, cap, next;
}g[N*N*2];
int head[N], pre[N], cur[N], gap[N], level[N];
int cnt, nv, cas = 0;
int arr[110], brr[110];
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++;
}
int sap(int s, int t)
{
    memset(level, 0, sizeof level);
    memset(gap, 0, sizeof gap);
    memcpy(cur, head, sizeof head);
    gap[0] = nv;
    int v = pre[s] = s, flow = 0, aug = INF;
    while(level[s] < nv)
    {
        bool flag = false;
        for(int &i = cur[v]; i != -1; i = g[i].next)
        {
            int u = g[i].to;
            if(g[i].cap > 0 && level[v] == level[u] + 1)
            {
                flag = true;
                pre[u] = v;
                v = u;
                aug = min(aug, g[i].cap);
                if(v == t)
                {
                    flow += aug;
                    while(v != s)
                    {
                        v = pre[v];
                        g[cur[v]].cap -= aug;
                        g[cur[v]^1].cap += aug;
                    }
                    aug = INF;
                }
                break;
            }
        }
        if(flag) continue;
        int minlevel = nv;
        for(int i = head[v]; i != -1; i = g[i].next)
        {
            int u = g[i].to;
            if(g[i].cap > 0 && minlevel > level[u])
                minlevel = level[u], cur[v] = i;
        }
        if(--gap[level[v]] == 0) break;
        level[v] = minlevel + 1;
        gap[level[v]]++;
        v = pre[v];
    }
    return flow;
}
int main()
{
    int t, l, w, n, d;
    scanf("%d", &t);
    while(t--)
    {
        cnt = 0;
        memset(head, -1, sizeof head);
        scanf("%d%d%d%d", &l, &w, &n, &d);
        for(int i = 1; i <= n; i++)
            scanf("%d%d", &arr[i], &brr[i]);
        nv = n*2 + 2;
        for(int i = 1; i <= n; i++)
        {
            add_edge(i, i + n, 1); //拆点连边，容量1
            if(brr[i] - d <= 0) //与汇点连边，容量1
                add_edge(0, i, 1);
            if(brr[i] + d >= w) //与源点连边，容量1
                add_edge(i + n, nv - 1, 1);
            for(int j = i + 1; j <= n; j++) //两机器人之间距离小于2d的连边，容量1
                if((arr[i] - arr[j]) * (arr[i] - arr[j]) + (brr[i] - brr[j]) * (brr[i] - brr[j]) <= d * d * 4)
                    add_edge(i + n, j, 1), add_edge(j + n, i, 1);
        }
        printf("Case %d: %d\n", ++cas, sap(0, nv - 1));
    }
	return 0;
}