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

 

计蒜客--引爆炸弹 并查集_第1张图片

计蒜客--引爆炸弹 并查集_第2张图片 题目链接 https://www.jisuanke.com/course/709/36599

题解:刚开始用的dp,然后有3组测试挂了,才发现用的dp考虑漏了一种情况。后来去网上才看到说要用并查集,这题并查集的思想就是把横纵坐标归一,以横坐标或者纵坐标(偏移后的)为数字进行并查集,具体思路看代码不难理解

 

#include 
#include 
#include 
#define MAXN 1010
using namespace std;
int n,m;
bool visited[2*MAXN];
int pre[2*MAXN];
char ch[MAXN][MAXN];

void Init() {
    for(int i = 0;i <= n+m+2;i++) pre[i] = i;
}

// 路径压缩
int Find(int x) {
    if(pre[x]==x) return x;
    int t = Find(pre[x]);
    pre[x] = t;
    return t;
}


void Union(int x,int y) {
    int xx = Find(x);
    int yy = Find(y);
    if(xx != yy) {
        pre[xx] = yy;
    }
    
    
}

int main() {
   // freopen("input.txt","r",stdin);
    
    scanf("%d%d",&n,&m);
    Init();
    for(int i = 0;i < n;i++) scanf("%s",ch[i]);
    
    for(int i = 0;i < n;i++)
        for(int j = 0;j < m;j++) {
            if(ch[i][j]=='1') {
                Union(i,j+n);
            }
        }
        
    int res = 0;
    
    for(int i = 0;i < n;i++)
        for(int j = 0;j < m;j++) {
            if(ch[i][j]=='1') {
                int t = Find(i);
                if(!visited[t]) res++,visited[t]=1;
                t = Find(j+n);
                if(!visited[t]) res++,visited[t]=1;    
            }
        }
        
    printf("%d\n",res);        
    
    return 0;
}

 

你可能感兴趣的:(图论)