poj 3498（枚举加最大流dinic）

这个题需要自己建一个超级源点，建边 s -> i -> i ' ； s->i 容量为 i 点企鹅的数量，i到 i ' 容量为能跳多少次，然后枚举到每一个点的最大流，如果等于企鹅的总数，记录这个点。。最后把所有能到达的点从小到大输出即可。。。。。

这里我刚开始的数组小了。。各种tle。。。。提醒一下吧。。。。

#include<iostream>
#include<cstdio>
#include<string>
#include<cmath>
using namespace std;
const int inf=1<<30;
const int maxn=100000;
#define cc(m,v) memset(m,v,sizeof(m))
struct node{
int u,v,f,next;
}edge[maxn];
int head[220],p,lev[220],cur[220];
int que[maxn];
void ainit(){
p=0,cc(head,-1);
}
bool bfs(int s,int t){
int i,u,v,qin=0,qout=0;
cc(lev,0),lev[s]=1,que[qin++]=s;
while(qout!=qin){
u=que[qout++];
for(i=head[u];i!=-1;i=edge[i].next)
if(edge[i].f>0 && lev[v=edge[i].v]==0){
lev[v]=lev[u]+1,que[qin++]=v;
if(v==t) return 1;
}
}
return lev[t];
}
int dinic(int s,int t){
int i,k,u,qin,f;
int flow=0;
while(bfs(s,t)){
memcpy(cur,head,sizeof(head));
u=s,qin=0;
while(1){
if(u==t){
for(k=0,f=inf;k<qin;k++)
if(edge[que[k]].f<f)
f=edge[que[i=k]].f;
for(k=0;k<qin;k++)
edge[que[k]].f-=f,edge[que[k]^1].f+=f;
flow+=f,u=edge[que[qin=i]].u;
}
for(i=cur[u];cur[u]!=-1;i=cur[u]=edge[cur[u]].next)
if(edge[i].f>0 && lev[u]+1==lev[edge[i].v]) break;
if(cur[u]!=-1)
que[qin++]=cur[u],u=edge[cur[u]].v;
else{
if(qin==0) break;
lev[u]=-1,u=edge[que[--qin]].u;
}
}
}
return flow;
}
void addedge(int u,int v,int f){
edge[p].u=u,edge[p].v=v,edge[p].f=f,edge[p].next=head[u],head[u]=p++;
edge[p].u=v,edge[p].v=u,edge[p].f=0,edge[p].next=head[v],head[v]=p++;
}
int main(){
int n,i,j,s,cas,sum,flag;
int ans[110],m,v,x[110],y[110];
double len;
scanf("%d",&cas);
while(cas--){
scanf("%d%lf",&n,&len);
ainit();
for(i=0,s=n<<1,sum=0;i<n;i++){
scanf("%d%d%d%d",&x[i],&y[i],&m,&v);
if(m) addedge(s,i,m),sum+=m;
addedge(i,i+n,v);
}
len=len*len;
for(i=0;i<n;i++) for(j=i+1;j<n;j++)
if(pow((double)x[i]-x[j],2)+pow((double)y[i]-y[j],2)<=len)
addedge(i+n,j,inf),addedge(j+n,i,inf);
for(i=0,flag=0;i<n;i++){
for(j=0;j<p;j+=2) edge[j].f+=edge[j^1].f,edge[j^1].f=0;
ans[i]=(dinic(s,i)==sum),flag+=ans[i];
}
if(!flag) printf("-1");
else{
for(i=0;i<n;i++) if(ans[i])
printf("%d ",i);
}
printf("\n");
}
return 0;
}