牛客练习赛47 E DongDong数颜色 （树上启发式合并）

链接：https://ac.nowcoder.com/acm/contest/904/E
来源：牛客网

DongDong数颜色
时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 524288K，其他语言1048576K
64bit IO Format: %lld
题目描述
DongDong是个喜欢数颜色的女孩子，她已经熟练地掌握了在序列上数颜色的操作，现在她开始学习如何在树上数颜色，现在给定一个n个点,n-1条边的树形图（视1号店为根），每个点有一个颜色，每次询问以x为根的子树中有多少种不同的颜色，DongDong轻松地解决了这个问题，但她想考考会编程的你。

输入描述:
第一行两个整数n,m

第二行n个整数，表示每个点的颜色

接下来n-1行每行u,v,表示存在一条从u到v的双向边（保证最终图形是树形图）

2<=n<=100000,1<=m,color<=n，
输出描述:
共m行：每行输出相应询问的答案
示例1
输入
复制
4 3
1 1 2 3
1 2
2 3
1 4
1
2
4
输出
复制
3
2
1

思路：

询问的都是子树的问题，很经典的dsu on tree，

和这题比较像：https://www.cnblogs.com/qieqiemin/p/11309973.html

把add 里维护答案的部分改一下即可。

细节见代码：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

转载于:https://www.cnblogs.com/qieqiemin/p/11310304.html