链接: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)个村庄,那么有水流过的村庄中有多少个村庄是"劳动力不足的村庄"。 两个数之间用空格隔开
示例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
说明
蓝色数字表示村民数量, 红色数字为所求答案
#include<set>
#include<map>
#include<list>
#include<queue>
#include<stack>
#include<math.h>
#include<vector>
#include<bitset>
#include<iomanip>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<bits/stdc++.h>
#define eps (1e-8)
#define MAX 0x3f3f3f3f
#define u_max 1844674407370955161
#define l_max 9223372036854775807
#define i_max 2147483647
#define re register
#define pushup() tree[rt]=max(tree[rt<<1],tree[rt<<1|1])
#define nth(k,n) nth_element(a,a+k,a+n); // 将 第K大的放在k位
#define ko() for(int i=2;i<=n;i++) s=(s+k)%i // 约瑟夫
#define ok() v.erase(unique(v.begin(),v.end()),v.end()) // 排序,离散化
using namespace std;
inline int read(){
char c = getchar(); int x = 0, f = 1;
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' & c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
typedef long long ll;
const double pi = atan(1.)*4.;
const int M=1e9;
const int N=1e5+5;
int n,k;
vector<int>mapp[200005],ans[200005];
int a[200005],v[200005],cut=0;
void dfs(int x){ // 从双向图变成单向图
v[x]=1;
for(int i=0;i<mapp[x].size();i++){
if(v[mapp[x][i]]) continue;
ans[x].push_back(mapp[x][i]);
dfs(mapp[x][i]);
}
return ;
}
int dfs1(int x){ // 对 <=k 的情况进行累和
for(int i=0;i<ans[x].size();i++)
v[x]+=dfs1(ans[x][i]);
if(a[x]<=k)
v[x]++;
return v[x];
}
int main(){
scanf("%d %d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
int m=n-1;
while(m--){
int x,y;
scanf("%d %d",&x,&y);
mapp[x].push_back(y);
mapp[y].push_back(x);
}
dfs(1);
memset(v,0,sizeof(v));
dfs1(1);
for(int i=1;i<=n;i++){
printf("%d ",v[i]);
}
return 0;
}