POJ 3498 && HDU 2334 March of the Penguins（最大流-Dinic）

Description
二维整数坐标系上有N块浮冰, 每块上有Ni只企鹅, 企鹅每次最多跳D个单位远, 浮冰有耐久度Mi, 每有一只企鹅从它上面起跳, 浮冰的耐久-1, 减到0浮冰就消失, 现在企鹅要跳到同一块浮冰上, 问企鹅可能最终聚集在哪些浮冰上
Input
第一行为用例组数T，每组用例第一行为一个整数N表示浮冰数和一个浮点数D表示企鹅最远跳跃长度，之后N行每行四个整数x,y,ni,mi分别表示浮冰的横纵坐标，浮冰上的企鹅数以及浮冰的耐久度
Output
对于每组用例，若最终所有企鹅可以停在某一浮冰上则输出浮冰编号，若有多块满足条件的浮冰则将其编号按从小到大的顺序输出，若不存爱满足条件的浮冰则输出-1
Sample Input
2
5 3.5
1 1 1 1
2 3 0 1
3 5 1 1
5 1 1 1
5 4 0 1
3 1.1
-1 0 5 10
0 0 3 9
2 0 1 1
Sample Output
1 2 4
-1
Solution
将点容量限制转化为边容量限制，拆点。源点向I连容量Ni的边，I向I’连容量Mi的边。对于I能在一步内到达的J，从I’向J连容量∞的边（这样I要经过J必须受Mj容量限制）。枚举汇点求最大流判可行性(即是否满流)
注意:题目浮冰编号为0~N-1，所以结果要减一
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
#define maxn 222
#define maxm 33333
#define INF 0x3f3f3f3f
int head[maxn],cur[maxn],d[maxn],st[maxm],s,e,no,n;
struct point
{
    int u,v,flow,next;
    point(){};
    point(int x,int y,int z,int w):u(x),v(y),next(z),flow(w){};
}p[maxm];
void add(int x,int y,int z)
{
    p[no]=point(x,y,head[x],z); 
    head[x]=no++;
    p[no]=point(y,x,head[y],0); 
    head[y]=no++;
}
void init()
{
    memset(head,-1,sizeof(head));
    no=0;
}
bool bfs()
{
    int i,x,y;
    queue<int>q;
    memset(d,-1,sizeof(d));
    d[s]=0; 
    q.push(s);
    while(!q.empty())
    {
        x=q.front();    
        q.pop();
        for(i=head[x];i!=-1;i=p[i].next)
        {
            if(p[i].flow&& d[y = p[i].v]<0)
            {
                d[y]=d[x]+1;
                if(y==e)    
                    return true;
                q.push(y);
            }
        }
    }
    return false;
}
int dinic()
{
    int i,loc,top,x=s,nowflow,maxflow=0;
    while(bfs()){
        for(i=s;i<=e;i++)   
            cur[i]=head[i];
        top=0;
        while(true)
        {
            if(x==e)
            {
                nowflow=INF;
                for(i=0;i<top;i++)
                {
                    if(nowflow>p[st[i]].flow)
                    {
                        nowflow=p[st[i]].flow;
                        loc=i;
                    }
                }
                for(i=0;i<top;i++)
                {
                    p[st[i]].flow-=nowflow;
                    p[st[i]^1].flow+=nowflow;
                }
                maxflow+=nowflow;
                top=loc;    
                x=p[st[top]].u;
            }
            for(i=cur[x];i!=-1;i=p[i].next)
                if(p[i].flow&&d[p[i].v]==d[x]+1) 
                    break;
            cur[x]=i;
            if(i!=-1)
            {
                st[top++]=i;
                x=p[i].v;
            }
            else 
            {
                if(!top)    
                    break;
                d[x]=-1;
                x=p[st[--top]].u;
            }
        }
    }
    return maxflow;
}
int N;
double D;
struct node
{
    int x,y,ni,mi;
}ice[maxn];
double dis[maxn][maxn];
bool check(int x)
{
    init();//初始化 
    s=0;//源点为0 
    e=2*N+1;//汇点为2*N+1 
    n=2*N;
    int sum=0;
    for(int i=1;i<=N;i++)//算出企鹅总数 
        if(i!=x)
            sum+=ice[i].ni;
    for(int i=1;i<=N;i++)//源点到每块浮冰建容量为其容纳量的边 
        if(i!=x)
            add(s,i,ice[i].ni);
    for(int i=N+1;i<=2*N;i++)//若该块浮冰与汇点距离不大于D则建容量为无穷的边 
        if(i!=x+N&&dis[i-N][x]<=D)
            add(i,e,INF);
    for(int i=1;i<=N;i++)//每块浮冰拆点后的两点间建容量为其耐久度的边 
        if(i!=x)
            add(i,i+N,ice[i].mi);
    for(int i=1;i<=N;i++)//如两块浮冰距离不大于D,则从一块浮冰拆点后的点与另一块浮冰建容量为无穷的边 
        for(int j=1;j<=N;j++)
        {
            if(i==x||j==x)
                continue;
            if(dis[i][j]<=D)
                add(i+N,j,INF);
        }
    return dinic()==sum;//判断是否满流 
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%lf",&N,&D);
        for(int i=1;i<=N;i++)
            scanf("%d%d%d%d",&ice[i].x,&ice[i].y,&ice[i].ni,&ice[i].mi);
        for(int i=1;i<=N;i++)//求出各浮冰之间的距离矩阵 
            for(int j=1;j<=N;j++)
                dis[i][j]=sqrt((double)(ice[i].x-ice[j].x)*(ice[i].x-ice[j].x)+(ice[i].y-ice[j].y)*(ice[i].y-ice[j].y));
        int res=0,ans[maxn];
        for(int i=1;i<=N;i++)//枚举汇点 
            if(check(i))
                ans[res++]=i;
        if(res)//存在满足条件的浮冰 
        {
            for(int i=0;i<res;i++)
                printf("%d%c",ans[i]-1,i==res-1?'\n':' ');//题中浮冰编号为0~N-1 
        }
        else//不存在满足条件的浮冰 
            printf("-1\n");
    }
    return 0;
}