牛客小白月赛25

白魔法师

https://ac.nowcoder.com/acm/contest/5600/C

B、白魔法师

并查集
其实很容易知道,如果你处理到一棵树父节点的最大白色数量,子节点最大白色连通数量,那么对这个点的操作之后的答案,就很好找了。
现在问题回到,怎么去处理白色棋子在各个连通块里面的数量,这就要设计一个数据结构,并查集
我们输入一串字符,对应树中结点颜色,那么在不操作的情况下,每个连通块可以预处理出来。
如果子节点和自己都是白棋子,那么这里要比普通的并查集麻烦一点的就是,要把父节点有方向的去规定,吧连通块白棋子小的合并到大的里面去。这样可以保证同一个连通块只加一次,不会重复累加。具体可以手动去写写康康,很容易发现这个BUG的。
那么后面只要去枚举黑色的点更新答案即可,吧各个方向上的白连通块数量加起来,)就是这个地方会累加出问题,如果随意合并集合的话。
最后还要注意没有黑色的情况,直接输出n就可以了。

Code

#include <bits/stdc++.h>
#pragma GCC optimize(2)
#pragma GCC optimize(3)
using namespace std;
#define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
typedef long long ll;
const ll MOD = 1e9 + 7;
inline ll read() { ll s = 0, w = 1; char ch = getchar(); while (ch < 48 || ch > 57) { if (ch == '-') w = -1; ch = getchar(); }    while (ch >= 48 && ch <= 57) s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();    return s * w; }
inline void write(ll x) { if (!x) { putchar('0'); return; } char F[200]; ll tmp = x > 0 ? x : -x; if (x < 0)putchar('-');    int cnt = 0;    while (tmp > 0) { F[cnt++] = tmp % 10 + '0';        tmp /= 10; }    while (cnt > 0)putchar(F[--cnt]); }
inline ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; }
ll qpow(ll a, ll b) { ll ans = 1;    while (b) { if (b & 1)    ans *= a;        b >>= 1;        a *= a; }    return ans; }    ll qpow(ll a, ll b, ll mod) { ll ans = 1; while (b) { if (b & 1)(ans *= a) %= mod; b >>= 1; (a *= a) %= mod; }return ans % mod; }
inline int lowbit(int x) { return x & (-x); }

const int N = 1e5 + 7;
vector<int> e[N];
int father[N];
char s[N];
int cnt[N], sz[N];

int find(int root) { //路径压缩
    int son = root;
    while (root != father[root])
        root = father[root];
    while (son != root) {
        int temp = father[son];
        father[son] = root;
        son = temp;
    }
    return root;
}

void merge(int a, int b) {
    int fa = find(a);
    int fb = find(b);
    if (cnt[fa] > cnt[fb]) {
        father[fb] = fa;
        cnt[fa] += cnt[fb] + 1;
    }
    else {
        father[fa] = fb;
        cnt[fb] += cnt[fa] + 1;
    }
}

int main() {
    int n = read();
    for (int i = 1; i <= n; ++i)    father[i] = i;
    scanf("%s", s + 1);
    for (int i = 1; i < n; ++i) {
        int u = read(), v = read();
        e[u].push_back(v);
        e[v].push_back(u);
        if (s[u] == 'W' && s[v] == 'W')    merge(u, v);
    }
    for (int i = 1; i <= n; ++i)
        if (s[i] == 'W')    sz[i] = cnt[father[i]] + 1;
    int ans = 0;
    for (int i = 1; i <= n; ++i)
        if (s[i] == 'B') {
            int sum = 1;
            for (auto it : e[i])
                if (s[it] == 'W')
                    sum += sz[it];
            ans = max(ans, sum);
        }
    if (!ans)    write(n);
    else    write(ans);
    return 0;
}
全部评论

相关推荐

点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务