当前位置: 代码迷 >> 综合 >> 紫书 习题 11-12 UVa 1665 (并查集维护联通分量)
  详细解决方案

紫书 习题 11-12 UVa 1665 (并查集维护联通分量)

热度:21   发布时间:2023-09-20 21:14:07.0

这道题要逆向思维

反过来从大到小枚举, 就是在矩阵中一点一点加进去数字,这样比较

好操作, 如果正着做就要一点一点删除数字, 不好做。

我们需要在这个过程中维护联通块的个数, 这里用到了并查集。

首先加进去一个数, 联通分量数字先加一, 然后再考虑有没有和其他联通分量

相连。从当前位置四个方向枚举, 如果这个数之前已经被选中, 同时

不是一个联通分量, 那么也就是说当前这个木块把两个联通分量变成一个

联通分量, 联通分量数减去一。

这里还要把二维的化为一维的编号, 方便并查集操作。

然后注意这里输出很奇怪, 最后输出一行的最后一个位置是要有空格的

不然会格式错误。


#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
using namespace std;const int MAXN = 1123456;
struct node
{int x, y, v;bool operator < (const node& rhs) const{return v < rhs.v;}
};
vector<node> nodes;
int a[MAXN], ans[MAXN], n, m, k, sum;
int vis[MAXN], f[MAXN];
int dir[4][2] = {0, 1, 0, -1, -1, 0, 1, 0};int find(int x)
{if(f[x] != x)f[x] = find(f[x]);return f[x];
}inline int ID(int x, int y) { return x * m + y; } //二维化一维 void add(node u)
{sum++;vis[ID(u.x, u.y)] = 1;REP(i, 0, 4) //判断是否和其他联通分量相连 {int xx = u.x + dir[i][0], yy = u.y + dir[i][1];if(0 <= xx && xx < n && 0 <= yy && yy < m && vis[ID(xx, yy)]) //vis的作用是之前这个数有没有被选过 {int a = find(ID(u.x, u.y)), b = find(ID(xx, yy));if(a != b) sum--, f[a] = b;}}
}int main()
{int T;scanf("%d", &T);while(T--){memset(vis, 0, sizeof(vis));nodes.clear();scanf("%d%d", &n, &m);REP(i, 0, n)REP(j, 0, m){int v;scanf("%d", &v);nodes.push_back(node{i, j, v});}scanf("%d", &k);REP(i, 0, k) scanf("%d", &a[i]);	sort(nodes.begin(), nodes.end()); //把点排序,这样加数比较方便 REP(i, 0, n * m) f[i] = i;int pos = nodes.size() - 1; sum = 0;for(int i = k - 1; i >= 0; i--){while(nodes[pos].v > a[i]) add(nodes[pos--]);ans[i] = sum;}REP(i, 0, k) printf("%d ", ans[i]); //注意最后一个数字后要有空格 printf("\n");}return 0;
}