POJ 8895 Percolation｜并查集

问题描述

总时间限制:  1000ms 内存限制:  32768kB

描述

定义一个N行N列的矩阵，矩阵中的每个元素是个方格，每个方格有两种可能的状态：开通的或关闭的。初始时，所有方格都是关闭的。输入数据的每一步会指定矩阵中某个原来关闭的方格变成开通的。要求编写程序判断当前是否存在从矩阵中最上面一行的任何一个开着的方格走到最下面一行的任何一个开着的方格的路径。如果存在的话，输出当前的步数。比如走到第14步时，矩阵变成上下通透的，那么就输出14。注意：输入数据中只会把矩阵中的一部分方格打开。如果所有步骤都执行完了，矩阵仍然不是上下通透的，那么输出-1。显然，矩阵变成上下通透的一个必要条件是：最上面一行和最下面一行都至少要有一个方格是打开的。

在矩阵中行走时，只能横着走或竖着走，不能斜着走，也不能走出矩阵的边界。

输入

输入的第一行是一个自然数T(1≤T≤10)，代表测试数据的组数。每组测试数据的第一行有两个自然数N和M，其中N（1≤N≤1,000）代表方阵的维度，M（1≤M≤N*N）代表本组测试中打开的方格数目。随后的M行中每行有两个自然数，分别代表所打开的方格的行、列下标。注意：本题中矩阵的下标从1开始，即所有下标的取值都是[1, N]区间中的正整数。

输出

每组测试数据输出一个自然数K，表示打开第K个方格后，矩阵变成上下通透的。如果M个方格都打开后，矩阵仍然不是上下通透的，那么输出-1。

样例输入

1
4 10 
2 2
3 1
4 2
4 4
1 2
2 3
2 1
3 2
3 4
3 1

样例输出

8

问题解决

这道题主要使用并查集解决，主要思路是在顶部和底部添加两个节点，一旦有第一行的全都连到上面的节点，一旦有最后一行的全部连到下面的节点，如果两个节点查一下是同一个集合的，则两者连通。

并查集主要就是写好两个函数，一个是findparent，用于查询最终的根节点，其中可以使用路径压缩增加查询效率；第二个是merge，用于合并两个集合，分别查找其根节点，然后把两者相统一即可。

#include <iostream>
#include <string.h>
using namespace std;

int parent[1002001];
int go[4][2]={
    -1,0,
    1,0,
    0,-1,
    0,1
};

int findparent(int x){
    if(parent[x]==x) return x;
    else return parent[x]=findparent(parent[x]);
}

void merge(int x, int y){
    int rootx=findparent(x);
    int rooty=findparent(y);
    if(rootx!=rooty){
        parent[rooty]=rootx;
    }
}
int main(){
    int T;
    cin>>T;
    while(T--){
        memset(parent,-1,sizeof(parent));
        parent[0]=0;parent[1]=1;
        int N,M,x,y,res=-1;
        bool changed=false;

        cin>>N>>M;
        for(int i=1;i<=M;i++){
            cin>>x>>y;
            if(changed==true) continue;
            parent[x*N+y]=x*N+y;
            if(x==1) merge(x*N+y,0);
            if(x==N) merge(x*N+y,1);
            for(int i=0;i<4;i++){
                int dx = x+go[i][0];
                int dy = y+go[i][1];
                if( dx>=1 && dx<=N && dy>=1 && dy<=N && (parent[dx*N+dy]!= -1)){
                    merge(dx*N+dy,x*N+y);
                }
            }
       
            if( findparent(0)==findparent(1) ){
                if(changed==false){
                    res=i;
                    changed=true;
                }
            }
        }
        cout<<res<<endl;
    }    
    return 0;
}