【HDOJ】4183 Pahom on Water

/* 4183 */
#include <iostream>
#include <string>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <vector>
#include <deque>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstring>
#include <climits>
#include <cctype>
#include <cassert>
#include <functional>
#include <iterator>
#include <iomanip>
using namespace std;
//#pragma comment(linker,"/STACK:102400000,1024000")

#define sti                set<int>
#define stpii            set<pair<int, int> >
#define mpii            map<int,int>
#define vi                vector<int>
#define pii                pair<int,int>
#define vpii            vector<pair<int,int> >
#define rep(i, a, n)     for (int i=a;i<n;++i)
#define per(i, a, n)     for (int i=n-1;i>=a;--i)
#define clr                clear
#define pb                 push_back
#define mp                 make_pair
#define fir                first
#define sec                second
#define all(x)             (x).begin(),(x).end()
#define SZ(x)             ((int)(x).size())
#define lson            l, mid, rt<<1
#define rson            mid+1, r, rt<<1|1

typedef struct {
int u, v, c, f;
} Edge_t;

const int maxn = 605;
Edge_t E[maxn*maxn];
vi vc[maxn];
double F[maxn];
int X[maxn], Y[maxn], R[maxn];
int pre[maxn], a[maxn], ID[maxn];
int n, m;

void addEdge(int u, int v, int c) {
E[m].u = u;
E[m].v = v;
E[m].c = c;
E[m].f = 0;

E[m+1].u = v;
E[m+1].v = u;
E[m+1].c = 0;
E[m+1].f = 0;

vc[u].pb(m);
vc[v].pb(m+1);

m += 2;
}

bool judge(int i, int j) {
if (F[i] >= F[j])
return false;
if ((R[i]+R[j])*(R[i]+R[j]) > (X[i]-X[j])*(X[i]-X[j])+(Y[i]-Y[j])*(Y[i]-Y[j]))
return true;
return false;
}

bool bfs(int s, int t) {
int u, v;
int sz, k;
queue<int> Q;

memset(a, 0, sizeof(a));
Q.push(s);
a[s] = INT_MAX;
pre[s] = s;

while (!Q.empty()) {
u = Q.front();
Q.pop();
sz = SZ(vc[u]);
rep(i, 0, sz) {
k = vc[u][i];
v = E[k].v;
if (!a[v] && E[k].f<E[k].c) {
a[v] = min(a[u], E[k].c-E[k].f);
pre[v] = u;
ID[v] = k;
Q.push(v);
}
}
}

return a[t]==0;
}

int EK(int s, int t) {
int ret = 0, tmp;
int u, v, k;

while (1) {
if (bfs(s, t))
break;

tmp = a[t];
for (v=t,u=pre[v]; v!=s; v=u,u=pre[v]) {
k = ID[v];
E[k].f += tmp;
E[k^1].f -= tmp;
}
ret += tmp;
}

return ret;
}

int main() {
ios::sync_with_stdio(false);
#ifndef ONLINE_JUDGE
freopen("data.in", "r", stdin);
freopen("data.out", "w", stdout);
#endif

int case_n, n_;
int s, t;
int ans;

scanf("%d", &case_n);
while (case_n--) {
scanf("%d", &n_);
n = n_ + n_;
m = 0;

rep(i, 0, n)
vc[i].clr();

rep(i, 0, n_) {
scanf("%lf %d %d %d", &F[i], &X[i], &Y[i], &R[i]);
if (F[i] == 400.0)
s = i;
else if (F[i] == 789.0)
t = i;
}

if (judge(s, t)) {
puts("Game is VALID");
continue;
}

rep(i, 0, n_) {
if (i!=s && i!=t)
addEdge(i<<1, i<<1|1, 1);
rep(j, 0, i) {
if (judge(i, j))
addEdge(i<<1|1, j<<1, 1);
else if (judge(j, i))
addEdge(j<<1|1, i<<1, 1);
}
}

addEdge(s<<1, s<<1|1, 2);

ans = EK(s<<1, t<<1);

if (ans >= 2)
puts("Game is VALID");
else
puts("Game is NOT VALID");
}

#ifndef ONLINE_JUDGE
printf("time = %d.\n", (int)clock());
#endif

return 0;
}