题目描述
在靠近南极的某处,一些企鹅站在许多漂浮的冰块上。由于企鹅是群居动物,所以它们想要聚集到一起,在同一个冰块上。企鹅们不想把自己的身体弄湿,所以它们在冰块之间跳跃,但是它们的跳跃距离,有一个上限。
随着气温的升高,冰块开始融化,并出现了裂痕。而企鹅跳跃的压力,使得冰块的破裂加速。幸运的是,企鹅对冰块十分有研究,它们能知道每块冰块最多能承受多少次跳跃。对冰块的损害只在跳起的时候产生,而落地时并不对其产生伤害。
现在让你来帮助企鹅选择一个冰面使得它们可以聚集到一起。
Input
输入
第一行整数N,和浮点数D,表示冰块的数目和企鹅的最大跳跃距离。
(1≤N ≤100) (0 ≤D ≤100 000),
接下来N行,xi, yi, ni and mi,分别表示冰块的X和Y坐标,该冰块上的企鹅数目,以及还能承受起跳的次数。
输出
输出所有可能的相聚冰块的编号,以0开始。如果不能相遇,输出-1。
样例输入
5 3.5
1 1 1 1
2 3 0 1
3 5 1 1
5 1 1 1
5 4 0 1
样例输出
1 2 4
提示
样例2: 3 1.1 -1 0 5 10 0 0 3 9 2 0 1 1
-1
题解
将i拆成i和n+i,n+i是可起跳点,i是可跳到的点。。
代码
#include<bits/stdc++.h>
#define N 205
#define M 100005
#define inf 100000000
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
double D;
int n,tot,S,T,cnt,sum;
int Head[N],ret[M],Next[M],len[M],pre[M];
int dis[N],x[N],y[N],ans[N];
inline void ins(int u,int v,int l)
{
ret[++tot]=v;len[tot]=l;pre[tot]=l;
Next[tot]=Head[u];Head[u]=tot;
}
double cal(int i,int j){return (x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);}
void build()
{
for (int i=1;i<n;i++)
for (int j=i+1;j<=n;j++)
if (cal(i,j)<=D*D)
{
ins(i+n,j,inf);ins(j,i+n,0);
ins(j+n,i,inf);ins(i,j+n,0);
}
}
bool bfs()
{
for (int i=1;i<=S;i++) dis[i]=-1;queue<int>q;
dis[S]=0;q.push(S);
while (!q.empty())
{
int now=q.front();q.pop();
for (int i=Head[now];i;i=Next[i])
{
if (dis[ret[i]]!=-1||len[i]==0) continue;
dis[ret[i]]=dis[now]+1;
if (ret[i]==T) return 1;
q.push(ret[i]);
}
}
return 0;
}
int dfs(int u,int aug)
{
if (u==T||aug==0) return aug;
int flow=0;
for (int i=Head[u];i;i=Next[i])
{
if (dis[ret[i]]!=dis[u]+1) continue;
int t=dfs(ret[i],min(aug-flow,len[i]));
len[i]-=t;len[i^1]+=t;flow+=t;
if (aug==flow) return flow;
}
if (flow==0) dis[u]=-1;
return flow;
}
int dinic()
{
int res=0;
while (bfs())
{
int flow=dfs(S,inf);
while (flow)
{
res+=flow;
flow=dfs(S,inf);
}
}
return res;
}
int main()
{
n=read();scanf("%lf",&D);tot=1;S=2*n+1;
for (int i=1;i<=n;i++)
{
x[i]=read();y[i]=read();
int w=read(),c=read();
sum+=w;
ins(S,i,w);ins(i,S,0);
ins(i,i+n,c);ins(i+n,i,0);
}
build();
for (T=1;T<=n;T++)
{
for (int i=1;i<=tot;i++) len[i]=pre[i];
if (dinic()==sum) ans[++cnt]=T-1;
}
if (cnt==0) printf("-1");
else
{
for (int i=1;i<cnt;i++) printf("%d ",ans[i]);
printf("%d",ans[cnt]);
}
return 0;
}