牛客小白月赛25 C 白魔法师并查集

链接：https://ac.nowcoder.com/acm/contest/5600/C
来源：牛客网

题目描述
你是一个白魔法师。
现在你拿到了一棵树，树上有个点，每个点被染成了黑色或白色。
你可以释放一次魔法，将某个点染成白色。（该点不一定是黑色点，也可以是白色点）
现在释放魔法后要保证最大的白色点连通块尽可能大。请求出最大白色连通块的大小。
注：所谓白色连通块，指这颗树的某个连通子图，上面的点全部是白色。
输入描述:

第一行输入一个正整数，代表树的顶点数量。
第二行输入一个长度为的、仅由’W’和’B’组成的字符串，第个点为’W’代表该点为白色，'B’代表该点为黑色。
接下来的行，每行输入两个正整数和，代表点和点有一条边连接。

输出描述:

一个正整数，代表施放魔法后，最大的白色连通块的大小。

示例1
输入
复制

4
WBBW
1 2
2 3
3 4

输出
复制

给定一个树,树上有黑节点和白节点,现在你可以修改一个点为白色,问修改哪个点
可以使得修改后白色的联通块最大?

用并查集维护联通块的节点个数
枚举每个黑点UUU,则修改UUU会使得UUU的白色联通块个数变成1+∑白色相邻节点的联通块节点个数1+\sum{白色相邻节点的联通块节点个数}1+∑白色相邻节点的联通块节点个数
求个MaxMaxMax即可

注意 :我以前用的并查集模板路径压缩是错的,居然还一直没有发现

//错误的路径压缩
int find(int x) {int ret = x;while(pre[ret] != -1)ret = pre[ret];if(ret != x) pre[x] = ret;return ret;
}

正确的路径压缩如下

int find(int x){return f[x]==x?x:f[x]=find(f[x]);
}

感谢 7QQQ**** 大佬的指正

完整代码如下

#define debug
#ifdef debug
#include <time.h>
#include "/home/majiao/mb.h"
#endif#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <math.h>
#define MAXN (100005)
#define int long long int
#define INF (0x7f7f7f7f)
#define QAQ (0)#pragma GCC optimize(2)using namespace std;#ifdef debug
#define show(x...) \do { \cout << "\033[31;1m " << #x << " -> "; \err(x); \} while (0)
void err() { cout << "\033[39;0m" << endl; }
#endiftemplate<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }int n, m, Q, K, pre[MAXN], sum[MAXN];vector<int> G[MAXN];
char color[MAXN];#if 0
//我常用的并查集模板
//今天发现写这道题TLE了   这是个大问题   这是个大问题
//这个是错误的路径压缩这个是错误的路径压缩这个是错误的路径压缩这个是错误的路径压缩
//
int find(int x) {int ret = x;while(pre[ret] != -1)ret = pre[ret];if(ret != x) pre[x] = ret;return ret;
}#else#define f pre
int find(int x){return f[x]==x?x:f[x]=find(f[x]);
}
#endif#if 0
void union_xy(int x, int y) {int rx = fa(x), ry = fa(y);if(rx != ry) {sum[rx] += sum[ry];pre[ry] = rx;}
}
#endif
void init() {//  memset(pre, -1, sizeof(pre));for(int i=1; i<=n; i++) pre[i] = i;
}signed main() {#ifdef debugfreopen("test", "r", stdin);clock_t stime = clock();
#endifscanf("%lld %s ", &n, color+1);init();for(int i=1, u, v; i<n; i++) {scanf("%lld %lld ", &u, &v);G[u].push_back(v), G[v].push_back(u);if(color[u] == color[v] && color[u] == 'W') {int rx = find(u), ry = find(v);pre[rx] = ry;}}int ans = 0;for(int i=1; i<=n; i++) {int rx = find(i);sum[rx] ++;ans = max(ans, sum[rx]);}for(int i=1; i<=n; i++) {if(color[i] == 'B') {int tmax = 1;for(auto v : G[i]) {if(color[v] == 'W') tmax += sum[find(v)];}ans = max(tmax, ans);}}printf("%lld\n", ans ? ans : n);#ifdef debugclock_t etime = clock();printf("rum time: %lf 秒\n",(double) (etime-stime)/CLOCKS_PER_SEC);
#endifreturn 0;
}

牛客小白月赛25 C 白魔法师并查集相关推荐

牛客小白月赛25 C.白魔法师
牛客小白月赛25 C.白魔法师题目链接题目描述你是一个白魔法师. 现在你拿到了一棵树,树上有个点,每个点被染成了黑色或白色. 你可以释放一次魔法,将某个点染成白色.(该点不一定是黑色点,也可以 ...
牛客小白月赛25 C白魔法师 (bfs | 并查集)
题目描述:题目链接题意分析: n个节点 n-1条边,非常明显的树形结构(题目中就有说 ),因此不需要考虑环的情况. 树上有两种颜色,一种是白色一种是黑色,我们可以推断出这题应该是考虑连通块的大小,那 ...
牛客小白月赛2 D. 虚虚实实(欧拉回路,并查集)
题目描述震为雷,临危不乱,亨通畅达:巽为风,柔顺伸展,厚载万物. 震卦:洊雷,震,君子以恐惧修省.一口金钟在淤泥,人人拿着当玩石,忽然一日钟悬起,响亮一声天下知. 巽卦:随风,巽,君子以申命行事.一 ...
牛客小白月赛25 D.抽卡
牛客小白月赛25 D.抽卡题目链接题目描述王子连接的国服终于上线啦~ 已知王子连接的抽卡系统如下: 共有 n 个卡池,第个卡池共有 aia_iai 种卡,每张卡的出货率都是相等的(也就是说该 ...
白魔法师--图的连通块问题（牛客小白月赛25）
链接:题目链接来源:牛客网白魔法师时间限制:C/C++ 1秒,其他语言2秒空间限制:C/C++ 262144K,其他语言524288K 64bit IO Format: %lld 题目描述你 ...
白魔法师（牛客小白月赛25 图、并查集）
白魔法师链接:https://ac.nowcoder.com/acm/contest/5600/C 题目描述你是一个白魔法师. 现在你拿到了一棵树,树上有 nnn 个点,每个点被染成了黑色或白色. ...
牛客小白月赛25 补题+题解[A-J]
加油加油加油! 文章目录 A.AOE还是单体? B.k-size字符串 C.白魔法师 D.抽卡 E.点击消除 F.疯狂的自我检索者 G.解方程 H.神奇的字母(二) I.十字爆破 J.异或和之和 A. ...
白魔法师-牛客小白月赛25
对于这个题目,我的做法是换根dp, 因为可以选择将某一个点染成白色(不论之前的点是黑是白) , 做法如下: 将每个点当成根节点, 然后将根节点染成白色, 完全符合题意:: 首先dfs一下, 自底向上, ...
牛客小白月赛25 C-白魔法师 ( 图论 + 并查集 )
题目链接解题报告: 思路:如果将一个黑色点染成白色,那么将得到一个白色连通块,这个连通块由和这个黑色点连结的所有白色连通块组成. 如果将一个白色点染成白色,那么不会有任何变化. 所以我们可以先并查集 ...

牛客小白月赛25 C 白魔法师并查集

牛客小白月赛25 C 白魔法师并查集相关推荐

最新文章

热门文章

牛客小白月赛25 C 白魔法师 并查集

牛客小白月赛25 C 白魔法师 并查集相关推荐

最新文章

热门文章

牛客小白月赛25 C 白魔法师并查集

牛客小白月赛25 C 白魔法师并查集相关推荐