luogu#P7159. 「dWoi R1」Sweet Fruit Chocolate

「dWoi R1」Sweet Fruit Chocolate

题目背景

事情来源 ……

我们的东条妈妈在残害了梦野之后,还想继续她的事业,残害可怜的最原。她发现最原很喜欢吃甜甜的巧克力,另外还有一种东西叫做西西弗水果,他富含营养但没那么好吃,因此,她想把这些巧克力淋在西西弗水果上。

题目描述

东条把想淋的巧克力做成了一个巧克力喷泉树。巧克力喷泉树是一棵 nn 个节点的树。每个节点都有一个西西弗水果。对于每一个节点 uu,你有两种选择:你可以在节点 uu 放置 aua_u 个水果,也可以一个水果都不放。然后,东条会在根节点往下淋巧克力汁。节点 uu 给最原带来的营养值是 uu 及其子树中所放置的西西弗水果的数量。东条想要知道,对于所有 2n2^n 个放水果方案,最原所获得的营养值之和的总和是多少。答案对 998244353998244353 取余。

树的根节点为 11

输入格式

第一行一个正整数 nn

第二行 nn 个正整数 aia_i

后面 n1n-1 行每行两个正整数 u,vu,v (1u,vn)(1\le u,v\le n),代表有树中一条边 (u,v)(u,v)

输出格式

一行,一个整数代表答案。

3
1 1 2
1 2
2 3
36

提示

样例 1 解释

SS 表示选中状态

  • S=000S=000 贡献 00
  • S=001S=001 贡献 11
  • S=010S=010 贡献 22
  • S=011S=011 贡献 33
  • S=100S=100 贡献 66
  • S=101S=101 贡献 77
  • S=110S=110 贡献 88
  • S=111S=111 贡献 99

数据规模与约定

对于 20%20\% 的数据,满足 n20n\le 20

对于另外 30%30\% 的数据,满足 u=v1u=v-1

对于 100%100\% 的数据,满足 2n1062\le n\le 10^61ai1091 \le a_i \le 10^9