poj3498：March of the Penguins——题解

最近的题解的故事背景割。

题目：
描述

在靠近南极的某处，一些企鹅站在许多漂浮的冰块上。由于企鹅是群居动物，所以它们想要聚集到一起，在同一个冰块上。企鹅们不想把自己的身体弄湿，所以它们在冰块之间跳跃，但是它们的跳跃距离有一个上限。
随着气温的升高，冰块开始融化，并出现了裂痕。而企鹅跳跃的压力，使得冰块的破裂加速。幸运的是，企鹅对冰块十分有研究，它们能知道每块冰块最多能承受多少次跳跃。对冰块的损害只在跳起的时候产生，而落地时并不对其产生伤害。
现在让你来帮助企鹅选择一个冰面使得它们可以聚集到一起。

输入

第一行一个正数：数据个数，最多100个。之后：

第一行整数N，和浮点数D，表示冰块的数目和企鹅的最大跳跃距离。
(1≤N ≤100) (0 ≤D ≤100 000),
接下来N行，xi, yi, ni and mi,分别表示冰块的X和Y坐标，该冰块上的企鹅数目，以及还能承受起跳的次数。

输出

每个数据：

输出所有可能的相聚冰块的编号，以0开始。如果不能相遇，输出-1。

样例输入

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
样例输出

1 2 4
-1

攒的有点多先写一道。
简单点说：网络流的题，自己建一个源点，源点连向每个冰块，容量为该冰块企鹅数。
冰块分为入点和出点，入点到出点容量为起跳次数。
一个冰块能跳到另一个冰块，则前者出点连后者入点，容量INF，后者出点连前者入点，容量INF。
枚举汇点。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=205;
const int INF=214748364;
struct{
    int next;
    int to;
    int w;
    int cun;
}edge[maxn*maxn];
int head[maxn],cnt=-1;
void add(int u,int v,int w){//u起点v终点w容量
    cnt++;
    edge[cnt].to=v;
    edge[cnt].w=w;
    edge[cnt].next=head[u];
    edge[cnt].cun=w;
    head[u]=cnt;
}
int lev[maxn],cur[maxn];//lev层数，cur[i]为以i为起点的边的编号 
bool bfs(int m,int zong){//m为汇点,有zong个点 
    int dui[zong+1],r=0;//队列和右指针 
    for(int i=1;i<=zong;i++){//初始化 
        lev[i]=-1;
        cur[i]=head[i];
    }
    dui[0]=1,lev[1]=0;
    int u,v;//u起点v终点 
    for(int l=0;l<=r;l++){//左指针 
        u=dui[l];
        for(int e=head[u];e!=-1;e=edge[e].next){
            v=edge[e].to;
            if(edge[e].w>0&&lev[v]==-1){//1.能走 2.未分层 
                lev[v]=lev[u]+1;
                r++;
                dui[r]=v;//v入队 
                if(v==m)return 1;//分层完毕 
            }
        }
    }
    return 0;//无法分层 
}
int dinic(int u,int flow,int m){//u当前点，flow为下面的点能够分配多大的流量,m终点 
    if(u==m)return flow;//终点直接全流入
    int res=0,delta;//res实际流量 
    for(int &e=cur[u];e!=-1;e=edge[e].next){//'&'相当于cur[u]=e;即流满的边不会再被扫一次 
        int v=edge[e].to;
        if(edge[e].w>0&&lev[u]<lev[v]){//只能从低层往高层流
            delta=dinic(v,min(edge[e].w,flow-res),m);
            if(delta>0){//如果增广 
                edge[e].w-=delta;//正向边容量减少 
                edge[e^1].w+=delta;//反向边仍量增加（暗示退流） 
                res+=delta;//扩张流量增加 
                if(res==flow)break;//可流的都流完了，及时跳出 
            }
        }
    }
    if(res!=flow)lev[u]=-1;//没流完，说明以后不能从这个点流出任何流量，那就不需要这个点了 
    return res;
}
double suan(int a1,int b1,int a2,int b2){
    return sqrt(pow((a2-a1),2)+pow((b2-b1),2));
}
int a[101],b[101],c[101],e[101];
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        int n;
        double d;
        int sum=0;
        scanf("%d%lf",&n,&d);
        for(int i=1;i<=maxn;i++){
            head[i]=-1;
        }
        cnt=-1;
        for(int i=1;i<=n;i++){
            scanf("%d%d%d%d",&a[i],&b[i],&c[i],&e[i]);
            add(i*2,i*2+1,e[i]);
            add(i*2+1,i*2,0);
            sum+=c[i];
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i==j)continue;
                if(suan(a[i],b[i],a[j],b[j])<=d){
                    add(i*2+1,j*2,INF);
                    add(j*2,i*2+1,0);
                }
            }
        }
        for(int i=1;i<=n;i++){
            if(c[i]!=0){
                add(1,i*2,c[i]);
                add(i*2,1,0);
            }
        }
        int ok=0;
        int first=0;
        for(int i=1;i<=n;i++){
            int zui=i*2;
            int ans=0;
            while(bfs(zui,2*n+1)==1){
                ans+=dinic(1,INF,zui);
            }
            if(ans==sum){
                if(first!=0)printf(" ");
                printf("%d",i-1);//编号以0开始所以-1（才不是我懒得改了） 
                first=1;
                ok=1;
            }
            for(int i=0;i<=cnt;i++){
                edge[i].w=edge[i].cun;
            }
        }
        if(ok==0)printf("-1");
        printf("\n");
    } 
    return 0;
}