计蒜客-引爆炸弹(并查集)

在一个 n×m 的方格地图上,某些方格上放置着炸弹。手动引爆一个炸弹以后,炸弹会把炸弹所在的行和列上的所有炸弹引爆,被引爆的炸弹又能引爆其他炸弹,这样连锁下去。
计蒜客-引爆炸弹(并查集)_第1张图片
现在为了引爆地图上的所有炸弹,需要手动引爆其中一些炸弹,为了把危险程度降到最低,请算出最少手动引爆多少个炸弹可以把地图上的所有炸弹引爆。

参考题解
并查集的思想,只要记录行号和列号就可以了。然后用并查集并在一起,如果能引爆其他炸弹的话,那么一定祖先是一样的。就可以直接路径压缩来做到了。

#include 
using namespace std;
const int maxn = 1e3+5;
char maze[maxn][maxn];
int pre[maxn<<1];
bool vis[maxn<<1];
int n, m;

inline void InitPre(int _lenght)
{
    for(int i = 1; i <= _lenght; i++)   pre[i] = i;
}

inline int GetPre(int x)
{
    return x == pre[x] ? x : pre[x] = GetPre(pre[x]);
}

inline void JoinNode(int _child, int _parent)
{
    int pre_child = GetPre(_child);
    int pre_parent = GetPre(_parent);
    if(pre_child == pre_parent) return ;
    pre[pre_child] = pre_parent;
}

int main()
{
    scanf("%d%d", &n, &m);
    InitPre(n+m+2);
    for(int i = 0; i < n; i++) scanf("%s", maze[i]);
    for(int i = 0; i < n; i++)
        for(int j = 0; j < m; j++)
            if(maze[i][j] == '1')
                JoinNode(i, j+n);
    int result = 0;
    for(int i = 0; i < n; i++)
        for(int j = 0; j < m; j++)
            if(maze[i][j] == '1')
            {
                int temp = GetPre(i);
                if(vis[temp] == false)  result++, vis[temp] = true;
                temp = GetPre(j+n);
                if(vis[temp] == false)  result++, vis[temp] = true;
            }
    printf("%d\n", result);
    return 0;
}

你可能感兴趣的:(数据结构)