牛客网 - Company(BFS)

题目链接:https://ac.nowcoder.com/acm/contest/322/C
时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld

题目描述

在一个偏僻的大山里，一共有n个村庄，编号1~n，每个村庄都有一定数量的村民，其中只有1号村庄有水井，为了方便村民们日常用水，村民们一共修建了n-1根水管，保证每一村庄都能有水喝。因为水是从高流向低，所以我们知道1号村庄海拔最高，与1号村庄直接相连的村庄高度次之，以此类推。
对于每一个村庄，如果这个村庄的村民人数<=k, 我们称之为"劳动力不足的村庄"。
现在我们想知道，如果水井不修在1号村庄，而是修在任意一个村庄，那么有水流过的村庄中有多少个村庄是"劳动力不足的村庄"（包括有水井的村庄本身）。

输入描述:

第一行输入两个数n(1 <= n <= 200000)村庄的数量，以及劳动力的评估标准k(1 <= k <= 1e9)
第二行有n个数，表示这n个村庄的村民数(1 <= a <= 1e9)
之后有n-1行，每行两个数字u, v（1 <= u, v <= n) u, v两个村庄之间有一条水管。

输出描述:

输出n个数，输出假设水井修在第i(1 <= i <= n)个村庄，那么有水流过的村庄中有多少个村庄是"劳动力不足的村庄"。
两个数之间用空格隔开

输入

8 10
12 15 11 4 5 19 14 20
3 7
1 5
2 1
6 5
2 3
6 4
8 3
5 10
13 15 19 5 3
2 3
4 1
1 2
3 5

输出

2 0 0 1 2 1 0 0
2 1 1 1 1

说明

蓝色数字表示村民数量，红色数字为所求答案
样例1:

样例2:

解题思路

BFS+剪枝走一波。。

#include <stdio.h>
#include <string.h>
#define N 200010
struct edge {
    int u, v;
}e[2 * N];
int n, k, cnt = 0, dp[N], a[N], vis[N], f[N];
void Add(int u, int v)
{
    e[++cnt] = (edge){f[u], v};
    f[u] = cnt;
}
void dfs(int x)
{
    int tx, ty, ans = 0;
    if (dp[x] != -1)
        return ;
    vis[x] = 1;
    for (int i = f[x]; i; i = e[i].u)
    {
        int v = e[i].v;
        if (vis[v])
            continue;
        dfs(v);
        ans += dp[v];
    }
    if (a[x] <= k)
        ans++;
    dp[x] = ans;
    return ;
}
int main()
{
    int u, v;
    scanf("%d%d", &n, &k);
    memset(dp, -1, sizeof(dp));
    for (int i = 0; i < n; i++)
        scanf("%d", &a[i]);
    for (int i = 0; i < n - 1; i++)
    {
        scanf("%d%d", &u, &v);
        Add(u - 1, v - 1);
        Add(v - 1, u - 1);
    }
    dfs(0);
    for (int i = 0; i < n; i++)
        printf("%d ", dp[i]);
    printf("\n");
    return 0;
}