题目描述
给定一棵 $n$ 个节点的树,每个点有个喜欢程度,要求选一个联通块,并且这个联通块包含最大的喜欢程度的方案数,输出的方案数对 $998244353$ 取模。
输入
第一行包含一个整数 $n$ 表示这棵树的点数。
接下来一行包含 $n$ 个整数,第 $i$ 个数表示第 $i$个结点的喜欢程度 $val_i$。
接下来 $n-1$ 行,每行包含两个整数 $x,y$,表示结点 $x$ 和 $y$ 之间有一条边相连。
输出
输出一行一个数表示方案数对 $998244353$ 取模的值。
样例输入输出
输入#1
复制
5
1 1 1 1 1
1 2
2 3
3 4
4 5
输入#2
复制
5
0 1 0 1 0
1 2
2 3
3 4
4 5
提示
对于 $100\%$ 的数据,$1 \leq n \leq 10^5$。喜爱程度均为 $32$ 位带符号整数范围内的数。