杭电5652 India and China Origins

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5652

题目大意:给你一个用n*m个小方块组成的长方形矩阵,其中有一些小方块不能经过,左下角坐标为(0,0),给你Q组坐标,每组坐标会把可以经过的小方块变成不能经过的小方块,问从第几组坐标开始矩阵的上部就不能到达矩阵的下部,如果都能到达输出-1。

解题思路:(第一次知道这种题还能用并查集做),我们可以倒着想,先将Q组坐标对应的小方块变成不能经过的,然后一个一个去掉,判断去掉哪个点可以连通即可。

具体操作就是先将首先将Q组坐标对应的小方块变成不能经过的,建立两个结点,第一个结点与第一排能经过的结点合并为一个集合,第二个结点与最后一排能经过的结点合并为一个集合,然后遍历矩阵将能联通结点连通,最后判断,两个节点是否在一个集合,如果不在,就按照坐标把小方块变成可以经过,再将合并小方块周围的四个小方块合并到一个集合里,再判断两个结点是否在一个集合,如此循环即可。

#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <cassert>
#define RI(N) scanf("%d",&(N))
#define RII(N,M) scanf("%d %d",&(N),&(M))
#define RIII(N,M,K) scanf("%d %d %d",&(N),&(M),&(K))
#define mem(a) memset((a),0,sizeof(a))
using namespace std;
const int inf=1e9;
const int inf1=-1*1e9;
typedef long long LL;
char ma[505][505];
int ima[505][505];
int par[250005];
int rankk[250005];
int xx[4]= {1,0,-1,0};
int yy[4]= {0,1,0,-1};
vector< pair<int,int> > qq;
void init(int n)
{
    for(int i=0; i<n; i++)
    {
        par[i]=i;
        rankk[i]=0;
    }
}
int findd(int x)
{
    if(par[x]==x) return x;
    else return par[x]=findd(par[x]);
}

void unite(int x,int y)
{
    x=findd(x);
    y=findd(y);
    if(rankk[x]<rankk[y]) par[x]=y;
    else
    {
        par[y]=x;
        if(rankk[x]==rankk[y]) rankk[x]++;
    }
}
int main()
{
    int t;
    RI(t);
    while(t--)
    {
        mem(ima);
        int n,m,Q;
        RII(n,m);
        for(int i=0; i<n; i++)
            scanf("%s",ma[i]);
        for(int i=0; i<n; i++)
            for(int j=0; j<m; j++)
                ima[i][j]=ma[i][j]-'0';
        RI(Q);
        int ans=Q;
        while(Q--)
        {
            int x1,y1;
            RII(x1,y1);
            ima[x1][y1]=1;
            pair<int,int>ppp(x1,y1);
            qq.push_back(ppp);
        }

        init(n*m+5);
        for(int i=0; i<m; i++)
        {
            if(ima[0][i]==0) unite(0,i+1);
        }

        for(int i=1; i<n; i++)
            for(int j=0; j<m; j++)
            {
                if(ima[i][j]==0)
                {
                    if(i-1>=0&&ima[i-1][j]!=1)
                    {
                        unite(i*m+j+1,(i-1)*m+j+1);
                    }
                    if(j-1>=0&&ima[i][j-1]!=1)
                    {
                        unite(i*m+j+1,i*m+j-1+1);
                    }
                }
            }


        for(int i=0; i<m; i++)
        {
            if(ima[n-1][i]==0) unite(m*n+1,(n-1)*m+i+1);
        }

        if(findd(0)==findd(m*n+1))
            cout<<-1<<endl;
        else
        {
            for(int i=ans-1; i>=0; i--)
            {
                int x=qq[i].first;
                int y=qq[i].second;
                ima[x][y]=0;
                for(int j=0; j<4; j++)
                {
                    int x2=x+xx[j];
                    int y2=y+yy[j];
                    if(x2>=0&&x2<n&&y2>=0&&y2<m&&ima[x2][y2]==0)
                    {

                        unite(x2*m+y2+1,x*m+y+1);
                    }
                    else if(x2==-1)
                    {
                        unite(y2+1,0);
                    }
                    else if(x2==n)
                    {
                        unite(n*m+1,(n-1)*m+y2+1);
                    }
                }
                ans--;
                if(findd(0)==findd(m*n+1)) break;


            }
            cout<<ans+1<<endl;
        }
        qq.clear();
    }
    return  0;
}

你可能感兴趣的:(并查集,杭电5652)