手机站,wordpress木木主题,网站快速排名推荐,免费h5制作平台免费模板题目
小红拿到了一个无向图#xff0c;初始每人节点是白色#xff0c;其中有若干个节点被染成了红色。小红想知道#xff0c;若将 i 号节点染成红色#xff0c;当前的红色连块的数量是多少? 你需要回答i∈[1,n] 的答案。
定义#xff0c;若干节点组成一个红色连通块初始每人节点是白色其中有若干个节点被染成了红色。小红想知道若将 i 号节点染成红色当前的红色连块的数量是多少? 你需要回答i∈[1,n] 的答案。
定义若干节点组成一个红色连通块当且仅当它们都是红色节点且在该图上可以通过无向边互相到达这些可以连通的节点构成的最大集合为一个连通块。
代码
考察并查集建图
#include iostream
#include bits/stdc.h
using namespace std;
/*保存所有边当边的两端节点都为红色合并连通块遍历节点若为红输出当前连通块数量all若为白遍历其连接的红节点得出其连接的连通块数量xcnt_i all - x 1
*/
vectorvectorbool edges;
string colors; // 记录颜色
int cnt 0; // 连通块数量
class UnionFind {vectorint parents;
public:UnionFind(int n) {parents.resize(n 1);for(int i 1; i n; i) {parents[i] i;}}void Union(int x, int y) {int fx find(x);int fy find(y);if(fx ! fy) {parents[fy] fx;}return;}int find(int x) {if(parents[x] ! x) {parents[x] find(parents[x]);}return parents[x];}int cnt_red() {int num 0;for(int i 1; i parents.size(); i) {if(parents[i] i colors[i - 1] R)num;}return num;}
};int main() {int n, m;cin n m;UnionFind* uf new UnionFind(n);cin colors;edges.resize(n 1);for(int i 1; i n; i) {edges[i].resize(n 1);}int u, v;for(int i 0; i m; i) {cin u v;edges[u][v] true;edges[v][u] true;if(colors[u - 1] R colors[v - 1] R) {uf-Union(u, v);}}cnt uf-cnt_red();/*遍历节点若为红输出当前连通块数量all若为白遍历其连接的红节点得出其连接的连通块数量xcnt_i all - x 1*/for(int i 1; i n; i) {if(colors[i - 1] R) {cout cnt;}else {unordered_setintst; // 存储i点连接的连通块for(int j 1; j n; j) {if(edges[i][j] true colors[j - 1] R){st.insert(uf-find(j));}}cout cnt - st.size() 1;}if(i ! n) cout endl;}return 0;
}
/*
4 4
WRWR
1 2
2 3
3 4
1 41
2
1
2
*/
