题目描述
见LOJ。
网络流
建图裸跑。判断是否满流。
简单题。
#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef double db;
const int maxn=1000+10,maxm=500000+10,inf=1000000000;
int h[maxn],d[maxn],now[maxn],go[maxm*2],dis[maxm*2],fx[maxm*2],next[maxm*2];
struct dong{
int x,y,w,r;
} a[maxn],b[maxn];
bool bz[maxn],czy;
int i,j,k,l,r,s,t,n,m,mx,top,tot,tmp,num,ca;
void add(int x,int y,int z,int d){
go[++tot]=y;
dis[tot]=z;
fx[tot]=tot+d;
next[tot]=h[x];
h[x]=tot;
}
int getdis(int x1,int y1,int x2,int y2){
return (x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);
}
bool pd(int x,int y){
if (getdis(b[x].x,b[x].y,a[y].x,a[y].y)<=a[y].r*a[y].r) return 1;else return 0;
}
int dfs(int x,int flow){
bz[x]=1;
if (x==t){
tmp+=flow;
return flow;
}
int r=now[x],k;
while (r){
if (!bz[go[r]]&&dis[r]&&d[x]==d[go[r]]+1){
k=dfs(go[r],min(flow,dis[r]));
if (k){
dis[r]-=k;
dis[fx[r]]+=k;
now[x]=r;
return k;
}
}
r=next[r];
}
return now[x]=0;
}
bool change(){
int i,r,tmp=inf;
fo(i,s,t)
if (bz[i]){
r=h[i];
while (r){
if (!bz[go[r]]&&dis[r]&&d[go[r]]+1-d[i]<tmp) tmp=d[go[r]]+1-d[i];
r=next[r];
}
}
if (tmp==inf) return 0;
fo(i,s,t)
if (bz[i]) d[i]+=tmp;
return 1;
}
int main(){
scanf("%d",&ca);
while (ca--){
scanf("%d%d",&n,&m);
fo(i,1,n+m+2) h[i]=0;
tot=0;
fo(i,1,n) scanf("%d%d%d%d",&a[i].x,&a[i].y,&a[i].w,&a[i].r);
fo(i,1,m) scanf("%d%d%d",&b[i].x,&b[i].y,&b[i].w);
mx=a[1].w;
fo(i,1,m)
if (pd(i,1)) mx+=b[i].w,b[i].w=0;
s=1;t=n+m+2;
num=0;
fo(i,1,m)
if (b[i].w){
czy=0;
fo(j,2,n)
if (pd(i,j)){
czy=1;
break;
}
if (!czy) continue;
fo(j,2,n)
if (pd(i,j)){
add(i+1,j+m+1,inf,1);
add(j+m+1,i+1,0,-1);
}
add(s,i+1,b[i].w,1);
add(i+1,s,0,-1);
num+=b[i].w;
}
czy=1;
fo(i,2,n){
if (mx<a[i].w){
czy=0;
break;
}
add(i+m+1,t,mx-a[i].w,1);
add(t,i+m+1,0,-1);
}
if (!czy){
printf("qaq\n");
continue;
}
tmp=0;
fo(i,s,t) d[i]=0;
do{
fill(bz+s,bz+t+1,0);
fo(i,s,t) now[i]=h[i];
while (dfs(s,inf)) fill(bz+s,bz+t+1,0);
}while (change());
if (num==tmp) printf("ZQC! ZQC!\n");else printf("qaq\n");
}
}