ACM学习历程—Hihocoder 1291 Building in Sandbox（dfs && 离线 && 并查集）

http://hihocoder.com/problemset/problem/1291

前几天比较忙，这次来补一下微软笔试的最后一题，这题是这次微软笔试的第四题，过的人比较少，我当时在调试B题，没时间看这一题。不过打过之前一场BestCoder的应该都会有点思路，虽然BC那题是二维，这题是三维的，但是思路应该是一样的，没错，就是离线加并查集。

正过来考虑的时候，发现第一个要求相邻块是好处理的，但是第二个要求能否到达(1000, 1000, 1000)这个条件似乎比较难判断，当时BC上的题根据题意是可以二分加搜索，但是这题的条件是不能二分的。

离线并查集的话，比较好想（如果做过BC那题的话），就是先算上所有块，搜索一遍全图，把非块的用并查集联通起来。然后倒着来拆块，看要拆的块是否和边界（这里取了(0, 0, 1)这个边界非块）的非块联通（有公共的根），然后将拆掉块的地方和周围非块的地方联通。直到某一处的块与边界不联通，那么就是No。否则最后就是Yes。

本地用dfs爆栈了，改bfs应该可以解决。不过直接交上去，评测机用dfs不会爆栈。。。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <queue>
#include <map>
#include <set>
#include <string>
#include <vector>

using namespace std;

const int maxN = ;
const int maxX = ;
int ufs[**];
int n, x[maxN], y[maxN], z[maxN];
int xx[] = {-, ,  , ,  , };
int yy[] = { , , -, ,  , };
int zz[] = { , ,  , , -, };

int findRoot(int x)
{
    if (ufs[x] == x) return x;
    int fa = findRoot(ufs[x]);
    ufs[x] = fa;
    return fa;
}

void mergeUfs(int x, int y)
{
    int fx, fy;
    fx = findRoot(x);
    fy = findRoot(y);
    ufs[fx] = fy;
}

bool inSameUfs(int x, int y)
{
    int fx, fy;
    fx = findRoot(x);
    fy = findRoot(y);
    if (fx == fy) return true;
    else return false;
}

inline int Hash(int x, int y, int z)
{
    return x*maxX*maxX+y*maxX+z;
}

void dfs(int x, int y, int z)
{
    int t = Hash(x, y, z), tt;
    if (ufs[t] == - || ufs[t] != -) return;
    if (ufs[t] == -) ufs[t] = t;
    for (int i = ; i < ; ++i)
    {
        if (x+xx[i] <  || y+yy[i] <  || z+zz[i] <= ) continue;
        if (x+xx[i] >= maxX || y+yy[i] >= maxX || z+zz[i] >= maxX) continue;
        tt = Hash(x+xx[i], y+yy[i], z+zz[i]);
        if (ufs[tt] != -) continue;
        dfs(x+xx[i], y+yy[i], z+zz[i]);
        mergeUfs(t, tt);
    }
}

bool judge(int x, int y, int z)
{
    bool flag = false;
    int t, tt, to;
    for (int i = ; i < ; ++i)
    {
        tt = Hash(x+xx[i], y+yy[i], z+zz[i]);
        if (ufs[tt] == -) flag = true;
    }
    if (!flag) return false;

    flag = false;
    t = Hash(x, y, z);
    to = Hash(, , );
    ufs[t] = t;
    for (int i = ; i < ; ++i)
    {
        tt = Hash(x+xx[i], y+yy[i], z+zz[i]);
        if (ufs[tt] == -) continue;
        if (findRoot(to) == findRoot(tt)) flag = true;
        mergeUfs(t, tt);
    }
    return flag;
}

void input()
{
    memset(ufs, -, sizeof(ufs));
    int t;
    scanf("%d", &n);
    for (int i = ; i < n; ++i)
    {
        scanf("%d%d%d", &x[i], &y[i], &z[i]);
        t = Hash(x[i], y[i], z[i]);
        ufs[t] = -;
    }
    for (int i = ; i < maxX; ++i)
        for (int j = ; j < maxX; ++j)
        {
            t = Hash(i, j, );
            ufs[t] = -;
        }
    for (int k = ; k < maxX; ++k)
        for (int i = ; i < maxX; ++i)
            for (int j = ; j < maxX; ++j)
                dfs(i, j, k);
}

void work()
{
    for (int i = n-; i >= ; i--)
    {
        if (!judge(x[i], y[i], z[i]))
        {
            printf("No\n");
            return;
        }
    }
    printf("Yes\n");
}

int main()
{
    //freopen("test.in", "r", stdin);
    //freopen("test.out", "w", stdout);
    int T;
    scanf("%d", &T);
    for (int times = ; times <= T; ++times)
    {
        input();
        work();
    }
    return ;
}