[LibreOJ β Round]ZQC的游戏

题目描述

见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");
    }
}