HDU 5652 图论之并查集

点击打开链接

题意：上边是中国，下边是印度，黑点的部分不可以走，下面的矩阵1代表黑点不能走，然后给了Q，每一次将一个点变成黑点，即不能走，问最少多少次就可以完成

思路：如果有一条黑点连成的线，从走到右的说明我们达成了目的，那就好办了，每一个黑点可以与周围的8个黑点相连，这样我们在最左边建个汇点最右边建个源点，每次询问这两个点在不在一个集合就行了，用并查集轻松实现，只要两个点相连就放到一个集合里#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=500010;
int f[maxn];
int dir[8][2]={{1,0},{-1,0},{0,1},{0,-1},{-1,-1},{-1,1},{1,-1},{1,1}};
char str[510][510];
void init(){
for(int i=0;i<maxn;i++){
f[i]=i;
}
}
int find1(int x){
if(x!=f[x])
f[x]=find1(f[x]);
return f[x];
}
void unite(int a,int b){
int aa=find1(a);
int bb=find1(b);
if(aa==bb) return ;
f[aa]=bb;
}
int main(){
int T,n,m,ask,a,b;
scanf("%d",&T);
while(T--){
init();
scanf("%d%d",&n,&m);
int S=0,T=n*m+1;
for(int i=1;i<=n;i++) scanf("%s",str[i]+1);
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(j==1&&str[i][j]=='1') unite(S,(i-1)*m+j);
if(j==m&&str[i][j]=='1') unite(T,(i-1)*m+j);
if(str[i][j]=='1'){
for(int k=0;k<8;k++){
int xx=i+dir[k][0];
int yy=j+dir[k][1];
if(xx<1||xx>n||yy<1||yy>m||str[xx][yy]!='1') continue;
unite((xx-1)*m+yy,(i-1)*m+j);
}
}
}
}
int ans=-1;
if(find1(S)==find1(T)) ans=0;
scanf("%d",&ask);
for(int j=1;j<=ask;j++){
scanf("%d%d",&a,&b);
a++;b++;
if(ans!=-1) continue;
str[a][b]='1';
if(b==1) unite(S,(a-1)*m+b);
if(b==m) unite(T,(a-1)*m+b);
for(int i=0;i<8;i++){
int xx=a+dir[i][0];
int yy=b+dir[i][1];
if(xx<1||xx>n||yy<1||yy>m||str[xx][yy]!='1') continue;
unite((xx-1)*m+yy,(a-1)*m+b);
}
if(find1(S)==find1(T)) ans=j;
}
printf("%d\n",ans);
}
return 0;
}