| 2022年中国大学生程序设计竞赛女生专场 |
|---|
| Finished |
YahAHa 特别喜欢有颜色的树,他有一棵 $$$n$$$ 个点、以 1 为根的有根树,树上每个点有一个颜色。YahAHa 的幸运数字是 $$$k$$$ ,于是他想问问你,在这棵树中,以 $$$x$$$ 为根的子树内,与 $$$x$$$ 距离不超过 $$$k$$$ 的点中,一共出现了多少种不同的颜色。定义 $$$x$$$ 与 $$$y$$$ 两个点的距离为,$$$x$$$ 到 $$$y$$$ 的简单路径上边的个数。YahAHa 有很多次这样的询问想问你,他一共会问你 $$$m$$$ 次。
第一行两个数 $$$n(1\leq n\leq 10^5)$$$ 和 $$$k(0\leq k\leq n)$$$,分别表示有根树的节点个数和询问中的距离。
第二行 $$$n$$$ 个整数,其中第 $$$i$$$ 个整数 $$$c_i(1\leq c_i\leq n)$$$ 表示 $$$i$$$ 号节点的颜色。
接下来 $$$n-1$$$ 行,每行两个整数 $$$x$$$ 和 $$$y$$$ 表示 $$$x$$$ 和 $$$y$$$ 之间连有一条边。
接下来一行一个整数 $$$m(1\leq m\leq 10^5)$$$ 表示询问的次数。
接下来 $$$m$$$ 行每行一个整数 $$$x_i(1\leq x_i\leq n)$$$ ,表示询问以 $$$x_i$$$ 为根的子树。
共 $$$m$$$ 行,每行一个整数表示答案。
4 1 1 2 3 3 1 2 2 3 2 4 4 1 2 3 4
2 2 1 1
| Name |
|---|


