T9270 mjt树
题目背景
从前森林里有一棵很大的mjt树,树上有很多小动物。
题目描述
mjt树上有 n 个房间,第 i 个房间住着 ai 只第bi 种小动物。
这n个房间用n-1条路连接起来,其中房间1位mjt树的根。
现在每个房间x的小动物想知道,以房间x为根的mjt树中有多少只它们的同类.
输入输出格式
输入格式:
第一行一个整数n,表示房间数
接下来n行,每行两个整数ai,bi
再之后n-1,每行两个整数x、y,表示x和y之间有一条路径
输出格式:
一行n个数,第i个数表示以房间i为根的mjt树中bi种小动物有多少只,两个数之间用空格隔开
输入输出样例
输入样例#1: 复制
5
2 1
3 1
4 2
5 1
6 2
1 2
1 3
3 4
3 5
输出样例#1: 复制
10 3 10 5 6
说明
bi<=n<=100000,ai<=1000
by xjjppm.
dfs,把他的之前遍历的该编号的数+上子树的减去进入该节点之前遍历的
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn =200007;
struct T{
int a,b;
}tree[maxn];
struct node{
int v,next;
}edge[maxn];
int n,num=0,head[maxn],sum[maxn];
void add_edge(int a,int b) {
edge[++num].v=b;edge[num].next=head[a];head[a]=num;
}
int size[maxn],ans[maxn];
void dfs(int x,int f) {
ans[x]=size[tree[x].b];
size[tree[x].b]+=tree[x].a;
for(int i=head[x];i;i=edge[i].next) {
int v=edge[i].v;
if(v!=f) dfs(v,x);
}
ans[x]=size[tree[x].b]-ans[x];
}
int main() {
scanf("%d",&n);
for(int i=1;i<=n;++i) {
scanf("%d%d",&tree[i].a,&tree[i].b);
}
for(int a,b,j=1;j<n;++j) {
scanf("%d%d",&a,&b);
add_edge(a,b);
add_edge(b,a);
}
dfs(1,0);
for(int i=1;i<=n;i++) {
printf("%d ",ans[i]);
}
return 0;
}