P6118 [JOI 2019 Final] 珍しい都市

题目背景

JOI 2019 Final T5 由于数据点较多,本题只评测其中的部分数据。

题目描述

JOI 国有 $N$ 个城市,城市从 $1$ 到 $N$ 编号。这些城市被 $N-1$ 条双向道路连接,第 $i$ 条路连接两个城市 $A_i$ 和 $B_i$。从任何城市出发,可以到达所有城市。 JOI 国有些特产,每种特产的编号都在 $1$ 到 $M$ 之间(包括 $1$ 和 $M$),但是 $1$ 到 $M$ 的某些整数可能不代表 JOI 国的特产。JOI 国的每个城市都产一种特产。$j$ 城产的特产是 $C_j$。多个城市可能产相同的特产。 我们定义两个城市之间的距离为从一个城市到另一个城市需要经过的最少道路数,对于城市 $x$,我们定义城市 $y$($y\neq x$)是**独特的城市**当且仅当对于任何一个城市 $z$($z\neq x,z\neq y$),$x$ 与 $y$ 间的距离不等于 $x$ 与 $z$ 之间的距离。 JOI 国交通部部长 K 先生想知道对于城市 $j$ 的**独特的城市**一共能产多少种特产。 给出 JOI 国的道路信息与每个城市产的特产,写一个程序计算对于每个城市的**独特的城市**,一共能产多少种特产。

输入格式

输出格式

说明/提示

样例解释 $1$: 对于城市 $1$,它的独特城市是城市 $2,3$,城市 $2$ 产特产 $2$,城市 $3$ 产特产 $3$,一共产两种特产,因此答案是 $2$; 对于城市 $2$,没有独特城市,因此输出 $0$; 对于城市 $3$,它的独特城市是城市 $1$,城市 $1$ 产特产 ,因此答案是 $1$; 对于城市 $4$,它的独特城市是城市 $1,3$,城市 $1,3$ 均产特产 $1$,因此答案是 $1$; 对于城市 $5$,它的独特城市是城市 $1,3$,城市 $1,3$ 均产特产 $1$,因此答案是 $1$。 注意:没有城市产特产 $3$。 对于 $4\%$ 的数据,$N\le 2000$。 另有 $32\%$ 的数据,$M=1$。 另有 $32\%$ 的数据,$M=N,C_j=j(1\le j \le N)$。 对于 $100\%$ 的数据,$1\le N \le 2\times 10^5,1\le M,A_i,B_i \le N,A_i \neq B_i,1\le C_j \le M$。