并查集的应用

并查集可以有很多应用。比如权值并查集或带扩展域的并查集,可以来维护一些可传递的关系。当然,并查集最直接的应用,则是来维护集合的的连通关系。

第一题
题意:给出一幅图,以及一系列操作,每次操作删去一个点,及这个点相关的所有边,并输出删点之后的连通块个数。
思路:这道题把过程倒着来就好了,可以看成建图的过程,如果要新加一条边,这条边连接的两个点若是在不同的连通块内,连边后,联通快个数就++。

#include
#include
using namespace std;
const int MAXN = 4e5+5;

vector<int> v[MAXN];
vector<int> ans;
struct Edge{
     
    int u, v;
}edge[MAXN];
struct Point{
     
    int p;
}point[MAXN];
bool book[MAXN];
int n, m, k;

//并查集部分
int pre[MAXN];
int getFather(int u){
     
    if(pre[u] == u) return u;
    return pre[u] = getFather(pre[u]);
}
void merge(int u, int v){
     
    int fu = getFather(u);
    int fv = getFather(v);
    if(fu != fv) pre[fu] = fv;
}

int main() {
     
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> m;
    for(int i = 0; i <= n; ++i)
        pre[i] = i;
    for(int i = 1; i <= m; ++i){
     
        cin >> edge[i].u >> edge[i].v;
    }
    cin >> k;
    int x;
    for(int i = 1; i <= k; ++i){
     
        cin >> x;
        book[x] = true;
        point[i].p = x;
    }

    int tot = 0;

    for(int i = 1; i <= m; ++i){
     
        if(book[edge[i].u] == false && book[edge[i].v] == false){
      //直接加入并查集
            merge(edge[i].u, edge[i].v);
        } else{
      //建图
            v[edge[i].u].push_back(edge[i].v);
            v[edge[i].v].push_back(edge[i].u);
        }
    }

    for(int i = 0; i < n; ++i){
      //求出连通块数量
        if(getFather(i) == i) tot++;
    }
    tot -= k;
    ans.push_back(tot);
    for(int i = k; i >= 1; --i){
     
        int node = point[i].p;
        int len = v[node].size();
        for(int j = 0; j < len; ++j){
     
            if(getFather(node) != getFather(v[node][j]) && book[v[node][j]] == false){
     
                merge(node, v[node][j]);
                tot--;
            }
        }
        tot++;
        book[node] = false;
        ans.push_back(tot);
    }
    int len = ans.size();
    for(int i = len-1; i >= 0; --i){
     
        cout << ans[i] << endl;
    }
    return 0;
}

第二题
这题并查集有两个应用,因为每一个点都只能有一次从0变到1,所以我们对于每一行可以维护一个并查集,每个点指向右端第一个0,这样利用并查集“暴力修改”即可。然后总体的连通性再开一个并查集维护

#include
#include
using namespace std;
const int MAXN = 4e5+5;

int tot;
int n, m, q;
char mp[1005][1005];

//并查集维护每一行的连通情况,每个连通块最右端为0
int rt[1005][1005]; //right[i][j]代表第i行第j列的父亲,根节点指向右边第一个0
int getFirstZero(int row, int col){
     
    if(rt[row][col] == col)
        return col;
    return rt[row][col] = getFirstZero(row, rt[row][col]);
}
void link(int row, int u, int v){
     
    int fu = getFirstZero(row, u);
    int fv = getFirstZero(row, v);
    if(fu != fv) rt[row][fu] = rt[row][fv];
}

//并查集维护所有1块的连通情况
int pre[1005*1005];
int getFather(int u){
     
    if(pre[u] == u) return u;
    return pre[u] = getFather(pre[u]);
}
int merge(int u, int v){
      //顺便维护连通块个数
    int fu = getFather(u), fv = getFather(v);
    if(fu != fv){
     
        pre[fu] = pre[fv];
        tot--;
    }
}

int main(){
     
    ios::sync_with_stdio(0);
    cin.tie(0);

    //获取输入
    cin >> n >> m;
    for(int i = 1; i <= n; ++i){
     
        for(int j = 1; j <= m; ++j){
     
            cin >> mp[i][j];
        }
        mp[i][m+1] = '0';
    }
    
    //预处理每行连通块情况
    for(int i = 1; i <= n; ++i){
     
        for(int j = 1; j <= m+1; ++j){
     
            rt[i][j] = j;
        }
    }
    for(int i = 1; i <= n; ++i){
     
        for(int j = 1; j <= m+1; ++j){
     
            if(mp[i][j] == '1') link(i, j, j+1);
        }
    }

    //预处理总体连通情况
    for(int i = 1; i <= n; ++i)
        for(int j = 1; j <= m; ++j)
            pre[i*m+j] = i*m+j;
    for(int i = 1; i <= n; ++i){
     
        for(int j = 1; j <= m; ++j){
     
            if(mp[i][j] == '1'){
     
                if(j+1 <= m && mp[i][j+1] == '1') merge(i*m+j, i*m+j+1);
                if(i+1 <= n && mp[i+1][j] == '1') merge(i*m+j, (i+1)*m+j);
            }
        }
    }
    tot = 0;
    for(int i = 1; i <= n; ++i){
     
        for(int j = 1; j <= m; ++j){
     
            if(mp[i][j] == '1' && getFather(i*m+j) == i*m+j) tot++;
        }
    }

    //输入给定区域
    cin >> q;
    int x1, x2, y1, y2;
    while(q--){
     
        cin >> x1 >> y1 >> x2 >> y2;
        for(int r = x1; r <= x2; ++r){
      //对每一行检查
            int tmp = getFirstZero(r, y1);
            while(tmp <= y2){
     
                tot++;
                mp[r][tmp] = '1';
                link(r, tmp, tmp+1);
                if(r-1 >= 1 && mp[r-1][tmp] == '1') merge(r*m+tmp,(r-1)*m+tmp); //可以与上面连通
                if(r+1 <= n && mp[r+1][tmp] == '1') merge(r*m+tmp, (r+1)*m+tmp); //可以与下面连通
                if(tmp-1 >= 1 && mp[r][tmp-1] == '1') merge(r*m+tmp, r*m+tmp-1); //可以与左边连通
                if(tmp+1 <= m && mp[r][tmp+1] == '1') merge(r*m+tmp, r*m+tmp+1); //可以与右边连通
                tmp = getFirstZero(r, tmp);
            }
            
        }

        cout << tot << endl;
    }
    return 0;
}

你可能感兴趣的:(并查集的应用)