问题 5419 --块的计数

5419: 块的计数

题目描述

给定一棵 $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
输出#1 复制
15
输入#2 复制
5
0 1 0 1 0
1 2
2 3
3 4
4 5
输出#2 复制
12

提示

对于 $100\%$ 的数据,$1 \leq n \leq 10^5$。喜爱程度均为 $32$ 位带符号整数范围内的数。
序号 标题 作者 发表时间 费用 订购数 操作