U172646 【PKUWC2019 D1T2】你和虚树的故事

给 $n$ 个点的树和 $m$ 种颜色，每个点有颜色 $c[i]\in[1,m]$，定义点集的虚树为最小的包含所有点的子图（包括所有路径上的点，注意此处虚树定义和普通虚树有所不同）。 选出 $i$ 种颜色，求出每种颜色点集各自的虚树，统计满足【这些颜色点集各自虚树交非空】的方案数量 $\pmod{998244353}$，求 $i=1,2,...,m$ 的所有答案

第一行两个整数 $n,m$ 第二行 $n$ 个整数 $c[1...n]$ 接下来 $n-1$ 行每行2个整数 $(u,v)$

一行 $m$ 个数表示答案

对于所有数据，$1\le n,m\le 10^5$