poj 3498(最大流+拆点)

题目链接：http://poj.org/problem?id=3498

思路：首先设一个超级源点，将源点与各地相连，边容量为各点目前的企鹅数量，然后就是对每个冰块i进行拆点了（i,i+n)，边容量为能够接受的受损程度，这样就把点权问题转化为边权问题了，然后就是对于那些能够相互跳跃的冰块之间连边（i+n,j),(j+n,i)，边容量为inf。最后就是枚举汇点看是否等于总数。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<queue>
 #include<cmath>
 using namespace std;
 #define MAXN 222
 #define MAXM 2222222
 #define inf 1<<30

 struct Edge{
     int v,cap,next;
 }edge[MAXM];

 int n,NE,vs,vt,NV;
 double dist;
 int head[MAXN];

 void Insert(int u,int v,int cap)
 {
     edge[NE].v=v;
     edge[NE].cap=cap;
     edge[NE].next=head[u];
     head[u]=NE++;

     edge[NE].v=u;
     edge[NE].cap=;
     edge[NE].next=head[v];
     head[v]=NE++;
 }

 int level[MAXN],gap[MAXN];
 void bfs(int vt)
 {
     memset(level,-,sizeof(level));
     memset(gap,,sizeof(gap));
     level[vt]=;
     gap[level[vt]]++;
     queue<int>que;
     que.push(vt);
     while(!que.empty()){
         int u=que.front();
         que.pop();
         for(int i=head[u];i!=-;i=edge[i].next){
             int v=edge[i].v;
             if(level[v]!=-)continue;
             level[v]=level[u]+;
             gap[level[v]]++;
             que.push(v);
         }
     }
 }

 int pre[MAXN],cur[MAXN];
 int SAP(int vs,int vt)
 {
     bfs(vt);
     memset(pre,-,sizeof(pre));
     memcpy(cur,head,sizeof(cur));
     int maxflow=,aug=inf;
     int u=pre[vs]=vs;
     gap[]=NV;
     while(level[vs]<NV){
         bool flag=false;
         for(int &i=cur[u];i!=-;i=edge[i].next){
             int v=edge[i].v;
             if(edge[i].cap>&&level[u]==level[v]+){
                 flag=true;
                 pre[v]=u;
                 u=v;
                 aug=min(aug,edge[i].cap);
                 if(v==vt){
                     maxflow+=aug;
                     for(u=pre[v];v!=vs;v=u,u=pre[u]){
                         edge[cur[u]].cap-=aug;
                         edge[cur[u]^].cap+=aug;
                     }
                     aug=inf;
                 }
                 break;
             }
         }
         if(flag)continue;
         int minlevel=NV;
         for(int i=head[u];i!=-;i=edge[i].next){
             int v=edge[i].v;
             if(edge[i].cap>&&level[v]<minlevel){
                 minlevel=level[v];
                 cur[u]=i;
             }
         }
         if(--gap[level[u]]==)break;
         level[u]=minlevel+;
         gap[level[u]]++;
         u=pre[u];
     }
     return maxflow;
 }
 struct Point{
     double x,y;
     int num1,num2;
 }p[MAXN];

 double Get_dist(int i,int j)
 {
     double d1=(p[i].x-p[j].x)*(p[i].x-p[j].x);
     double d2=(p[i].y-p[j].y)*(p[i].y-p[j].y);
     return sqrt(d1+d2);
 }
 void Build()
 {
     NE=;
     memset(head,-,sizeof(head));
     vs=,NV=*n+;
     for(int i=;i<=n;i++)Insert(vs,i,p[i].num1);
     for(int i=;i<=n;i++)Insert(i,i+n,p[i].num2);
     for(int i=;i<=n;i++){
         for(int j=i+;j<=n;j++){
             if(Get_dist(i,j)<=dist){
                 Insert(i+n,j,inf);
                 Insert(j+n,i,inf);
             }
         }
     }
 }
 int ans[MAXN],cnt;
 int main()
 {
     int _case,sum;
     scanf("%d",&_case);
     while(_case--){
         scanf("%d%lf",&n,&dist);
         sum=;
         for(int i=;i<=n;i++){
             scanf("%lf%lf%d%d",&p[i].x,&p[i].y,&p[i].num1,&p[i].num2);
             sum+=p[i].num1;
         }
         cnt=;
         for(int i=;i<=n;i++){
             Build();
             if(SAP(vs,i)==sum)ans[cnt++]=i;
         }
         if(cnt){
             for(int i=;i<cnt;i++){
                 printf(i?" %d":"%d",ans[i]-);
             }
             printf("\n");
         }else
             puts("-1");
     }
     return ;
 }