Islands UVA - 1665 (并查集+优化)

最新推荐文章于 2021-11-30 06:01:53 发布

肘子zhouzi

最新推荐文章于 2021-11-30 06:01:53 发布

阅读量203

点赞数

文章标签： check and set

本文链接：https://blog.csdn.net/zhouzi2018/article/details/85266095

版权

传送门

题意：给出一个n*m的矩阵，每个格子都有一个[1,10^9]的正整数。再输入T个整数ti，对于大于ti的正整数组成多少个联通块。

题解：使用并查集，之后按照从大到小的顺序进行遍历Ti。

附上代码：


#include<bits/stdc++.h>
#define _for(i,a,b) for( int i=(a); i<(b); ++i)
#define _rep(i,a,b) for( int i=(a); i<=(b); ++i)
using namespace std;
inline int readint() { int x; scanf("%d", &x); return x; }

struct Point{
    int x,y,v;
    void init(int r, int c, int value){
        assert(r >= 0), assert(c >= 0);
        x = r, y = c, v = value;
    }
};

const int MAXM = 1024, MAXN = 1024, MAXT = 100000 + 5;
int m, n, T, t[MAXT], Ans[MAXT], indice[MAXN][MAXM], pa[MAXN*MAXM];
Point points[MAXN*MAXM];
vector<int> tmp; // 储存p的邻居
int findPa(int i) { return pa[i] == i ? i : (pa[i] = findPa(pa[i])); }
void getAdjs(const Point& p, vector<int>& ans) {
    ans.clear();
    int r = p.x-1, c = p.y;
    if(r>=0) ans.push_back(indice[r][c]);
    r=p.x+1;
    if(r<n) ans.push_back(indice[r][c]);
    r = p.x; c = p.y-1;
    if(c>=0) ans.push_back(indice[r][c]);
    c = p.y+1;
    if(c<m) ans.push_back(indice[r][c]);    
}

void solve() {
    scanf("%d%d", &n, &m);
    int psz = 0;
    _for(i, 0, n) _for(j, 0, m) points[psz++].init(i,j,readint());
    sort(points, points+psz, [](const Point& p1, const Point& p2){
        return p1.v > p2.v;
    });

    _for(i, 0, psz){
        const Point& p = points[i];
        pa[i] = i;
        indice[p.x][p.y] = i;
    }
    
    T = readint();
    _for(i, 0, T)  t[i] = readint();
   
    int pi = 0;
    Ans[T] = 0;
    for(int i = T-1; i>=0; i--){
        int& ans = Ans[i];
        ans = Ans[i+1];
        while(pi < psz && points[pi].v > t[i]){
            ans++;            
            getAdjs(points[pi], tmp); // 遍历p的上下左右四个邻居
            for(auto j : tmp){
                Point& pn = points[j];
                if(pn.v <= t[i]) continue;
                int pnpa = findPa(indice[pn.x][pn.y]);
                if (pnpa == findPa(pi)) continue;
                pa[pnpa] = pi;
                ans--;
            }
            pi++;
        }
    }
}

int main(){
    int Z = readint();
    _for(z, 0, Z) {
        solve();
        _for(i, 0, T) printf("%d ", Ans[i]);
        puts("");
    }
    return 0;
}