Company

题目描述 

在一个偏僻的大山里， 一共有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)个村庄，那么有水流过的村庄中有多少个村庄是"劳动力不足的村庄"。

两个数之间用空格隔开

示例1

输入

复制

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

输出

复制

2 0 0 1 2 1 0 0

说明

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

示例2

输入

复制

5 10
13 15 19 5 3
2 3
4 1
1 2
3 5

输出

复制

2 1 1 1 1

说明

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

思路：从1点开始dfs

#include <bits/stdc++.h>
using namespace std;
const int MAXN=201000;
struct node
{
    int u,v,next;
}a[2*MAXN];
int head[MAXN];
int cnt=1;
void add(int u,int v)
{
    a[cnt].u=u;
    a[cnt].v=v;
    a[cnt].next=head[u];
    head[u]=cnt++;
}
int ans[MAXN];
int b[MAXN];
int n,k;
void dfs(int s,int fa)
{
    if(b[s]<=k)
        ans[s]++;
    for(int i=head[s];i!=-1;i=a[i].next)
    {
        if(a[i].v==fa)
            continue;
        dfs(a[i].v,s);
        ans[s]+=ans[a[i].v];
    }
}
int main()
{
    scanf("%d%d",&n,&k);
    int u,v;
    memset(head,-1,sizeof(head));
    cnt=1;
    for(int i=1;i<=n;i++)
        scanf("%d",&b[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    memset(ans,0,sizeof(ans));
    dfs(1,0);
    for(int i=1;i<n;i++)
        printf("%d ",ans[i]);
    printf("%d\n",ans[n]);
    return 0;
}