POJ 3498 March of the Penguins(枚举+最大流)

题目链接：poj3498

题意:

       在X,Y坐标系中有N(N<=100)个冰块...有些冰块上有若干只企鹅..每只企鹅一次最多跳M距离..一个冰块在有Mi个企鹅离开..就会消失..问有哪些冰块可以作为集合点..就是所有企鹅都能成功到这个冰块上来.

分析:

       首先我们枚举每块冰,看看这块冰如果作为集合点,是否所有企鹅都能到这块冰上.

       建图:

把每块冰分成两个点i和i+110. i表示进入i冰块的点(可以有无数企鹅过来,所以从别的冰到i有边,容量为INF) i+n表示从i冰块出去的点(最多只能有Mi企鹅从这跳出去,所以从i到i+n有边,且容量为Mi)

从源点S到i有边(S, i, i点初始企鹅数).

从i到i+n有边(i, i+n, Mi). 表示第i块冰最多只有Mi个企鹅能跳走.

因为i+n表示的是第i个跳走的点,所以如果冰块i和j之间的距离<=企鹅能跳跃的距离M,有边(i+n, j, INF)

假设我们当前枚举第x块冰块作为集合点,那么(x分成x和x+110两个点)x点就是汇点(不是x+110点哦),我们只要计算到x点的流量是否==企鹅总数即可.

注意最后结果输出编号从0计数.

（每次枚举一个终点都要重新建图）

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<cmath>
#include<iostream>
#define INF 1e9
#define INFLL 1LL<<60
using namespace std;
const int maxn=500+10;

struct Edge
{
    int from,to,cap,flow;
    Edge(){}
    Edge(int f,int t,int c,int fl):from(f),to(t),cap(c),flow(fl){}
};

struct Dinic
{
    int n,m,s,t;
    vector<Edge> edges;
    vector<int> G[maxn];
    int d[maxn];
    int cur[maxn];
    bool vis[maxn];
    void init(int n,int s,int t)
    {
        this->n=n, this->s=s, this->t=t;
        edges.clear();
        for(int i=0;i<n;i++) G[i].clear();
    }

    void AddEdge(int from,int to,int cap)
    {
        edges.push_back( Edge(from,to,cap,0) );
        edges.push_back( Edge(to,from,0,0) );
        m=edges.size();
        G[from].push_back(m-2);
        G[to].push_back(m-1);
    }

    bool BFS()
    {
        queue<int> Q;
        memset(vis,0,sizeof(vis));
        vis[s]=true;
        d[s]=0;
        Q.push(s);
        while(!Q.empty())
        {
            int x=Q.front(); Q.pop();
            for(int i=0;i<G[x].size();i++)
            {
                Edge e=edges[G[x][i]];
                if(!vis[e.to] && e.cap>e.flow)
                {
                    vis[e.to]=true;
                    d[e.to] = d[x]+1;
                    Q.push(e.to);
                }
            }
        }
        return vis[t];
    }

    int DFS(int x,int a)
    {
        if(x==t || a==0) return a;
        int flow=0,f;

        for(int& i=cur[x];i<G[x].size();++i)
        {
            Edge& e=edges[G[x][i]];
            if(d[e.to]==d[x]+1 && (f=DFS(e.to, min(a,e.cap-e.flow) ) )>0 )
            {
                e.flow+=f;
                edges[G[x][i]^1].flow-=f;
                flow+=f;
                a-=f;
                if(a==0) break;
            }
        }
        return flow;
    }

    int Max_Flow()
    {
        int flow=0;
        while(BFS())
        {
            memset(cur,0,sizeof(cur));
            flow += DFS(s,INF);
        }
        return flow;
    }
}DC;
struct node
{
	int x, y;
	int num,have;
}s[200];
int n;//个数
double m;//距离
double dist[maxn][maxn];//dist[i][j]表示从牛棚i到牛棚j所需要耗费的时间

double getdis(int i,int j)
{
	double sum=sqrt((double)((s[i].x-s[j].x)*(s[i].x-s[j].x)+(s[i].y-s[j].y)*(s[i].y-s[j].y)));
	return sum;
}
int main()
{
	int cas;
	scanf("%d",&cas);
	while(cas--)
	{
		scanf("%d%lf",&n,&m);
		int sum=0;
		for(int i=0;i<n;i++)
		{
			scanf("%d%d%d%d",&s[i].x,&s[i].y,&s[i].num,&s[i].have);
			sum+=s[i].num;
		}
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
		{
			dist[i][j]=getdis(i,j);
		}


		int ans=0;
		int a[200];
		for(int i=0;i<n;i++)//枚举终点
		{
			DC.init(260,250,i);
			for(int i=0;i<n;i++)
			{
				DC.AddEdge(250,i,s[i].num);
				DC.AddEdge(i,i+110,s[i].have);
			}
			for(int i=0;i<n;i++)
				for(int j=0;j<n;j++)
					if(dist[i][j]<=m&&i!=j)
						DC.AddEdge(i+110,j,INF);//时间在限制时间内的  可以从i到j的道路
			if(DC.Max_Flow()==sum)
			{
				a[ans++]=i;
			}

		}
		if(ans==0)printf("-1\n");
		else
		{
			for(int i=0;i<ans;i++)
				printf("%d%c",a[i],i==(ans-1)?'\n':' ');
		}
	}
    return 0;
}