欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

牛客小白月赛25 C.白魔法师

程序员文章站 2022-03-13 17:09:17
...

牛客小白月赛25 C.白魔法师

牛客小白月赛25 C.白魔法师

主要思路: 

BFS求一下每个白色连通快的大小,然后对每一个黑色点逐个判断,判断通过这个黑色点可以连通多少个白色连通块,把这些白色连通块大小相加再+1,最后答案维护一下最大值就行了。

如果没有黑色点的话,直接输出最大的白色连通块就好。否则输出上述维护的ans。

这个题真滴就差一点,一开始犯浑想成求树上最大直径了,想成只能连接两个连通块,fo自己了,也算是个有点小难度的图论题吧。

看着题解很多都是并查集做的,本蒟蒻的比标程还稍微快那么一倍,O(∩_∩)O哈哈~

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <stack>
#include <map>
#include <sstream>
#include <cstring>
#include <set>
#include <cctype>
#include <bitset>
#define IO                       \
    ios::sync_with_stdio(false); \
    // cin.tie(0);                  \
    // cout.tie(0);
using namespace std;
typedef long long LL;
const int maxn = 2e5 + 10;
const int maxm = 1e6 + 10;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int inf = 0x3f3f3f3f;
const LL mod = 1e9 + 7;
int dis[8][2] = {0, 1, 1, 0, 0, -1, -1, 0, 1, -1, 1, 1, -1, 1, -1, -1};
char color[maxn];
int size[maxn];
int ID[maxn];
bool vis[maxn];
int n, k; // k 作为标号
int maxsize = 0;
vector<int> v[maxn];
void BFS(int s)
{
    queue<int> q;
    q.push(s);
    vis[s] = 1;
    ID[s] = k;
    int cnt = 1;
    while (!q.empty())
    {
        int u = q.front();
        q.pop();
        for (int i = 0; i < v[u].size(); i++)
        {
            int vv = v[u][i];
            if (!vis[vv] && color[vv] == 'W')
            {
                ID[vv] = k;
                cnt++;
                vis[vv] = 1;
                q.push(vv);
            }
        }
    }
    size[k] = cnt;
    maxsize = max(maxsize, cnt);
}

int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
#endif
    IO;
    int c = 0; // 记录黑色个数
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
    {
        scanf(" %c", &color[i]);
        if (color[i] == 'B')
            ++c;
    }
    int x, y;
    for (int i = 1; i <= n - 1; i++)
    {
        scanf("%d %d", &x, &y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    for (int i = 1; i <= n; i++)
        if (color[i] == 'W' && !vis[i])
        {
            ++k;
            BFS(i);
        }
    int ans = 0;
    for (int i = 1; i <= n; i++)
    {
        int t = 0;
        if (color[i] == 'B')
        {
            for (int j = 0; j < v[i].size(); j++)
            {
                if (color[v[i][j]] == 'W')
                    t += size[ID[v[i][j]]];
            }
        }
        ans = max(ans, t);
    }

    if (c == 0)
    {
        cout << maxsize;
    }
    else
    {
        cout << ans + 1;
    }
    return 0;
}

 

 

 

 

 

 

相关标签: 图论