#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #ifdef LOCAL # include "debug_print.hpp" # define debug(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__) #else # define debug(...) (static_cast(0)) #endif using namespace std; #define rep1(a) for(int i = 0; i < a; i++) #define rep2(i, a) for(int i = 0; i < a; i++) #define rep3(i, a, b) for(int i = a; i < b; i++) #define rep4(i, a, b, c) for(int i = a; i < b; i += c) #define overload4(a, b, c, d, e, ...) e #define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__) #define rrep1(a) for(int i = (a)-1; i >= 0; i--) #define rrep2(i, a) for(int i = (a)-1; i >= 0; i--) #define rrep3(i, a, b) for(int i = (b)-1; i >= a; i--) #define rrep4(i, a, b, c) for(int i = (b)-1; i >= a; i -= c) #define rrep(...) overload4(__VA_ARGS__, rrep4, rrep3, rrep2, rrep1)(__VA_ARGS__) #define ALL(v) v.begin(), v.end() #define RALL(v) v.rbegin(), v.rend() #define UNIQUE(v) v.erase( unique(v.begin(), v.end()), v.end() ); #define pb push_back using ll = long long; using D = double; using LD = long double; using P = pair; using vi = vector; using vl = vector; template using vc = vector; template using vvc = vector>; template using vvvc = vector>; template using vvvvc = vector>; template using vvvvvc = vector>; #define vv(type, name, h, ...) \ vector> name(h, vector(__VA_ARGS__)) #define vvv(type, name, h, w, ...) \ vector>> name( \ h, vector>(w, vector(__VA_ARGS__))) #define vvvv(type, name, a, b, c, ...) \ vector>>> name( \ a, vector>>( \ b, vector>(c, vector(__VA_ARGS__)))) template using PQ = priority_queue>; template using minPQ = priority_queue, greater>; templatebool chmax(T &a, const T &b) { if (abool chmin(T &a, const T &b) { if (b ostream &operator<<(ostream &os, const pair &p) { os << p.first << " " << p.second; return os; } template istream &operator>>(istream &is, pair &p) { is >> p.first >> p.second; return is; } template ostream &operator<<(ostream &os, const vector &v) { int s = (int)v.size(); for (int i = 0; i < s; i++) os << (i ? " " : "") << v[i]; return os; } template istream &operator>>(istream &is, vector &v) { for (auto &x : v) is >> x; return is; } void in() {} template void in(T &t, U &...u) { cin >> t; in(u...); } void out() { cout << "\n"; } template void out(const T &t, const U &...u) { cout << t; if (sizeof...(u)) cout << sep; out(u...); } void outr() {} template void outr(const T &t, const U &...u) { cout << t; outr(u...); } template struct Fp { long long val; constexpr Fp(long long v = 0) noexcept : val(v % mod) { if (val < 0) val += mod; } constexpr int getmod() { return mod; } constexpr Fp operator - () const noexcept { return val ? mod - val : 0; } constexpr Fp operator + (const Fp& r) const noexcept { return Fp(*this) += r; } constexpr Fp operator - (const Fp& r) const noexcept { return Fp(*this) -= r; } constexpr Fp operator * (const Fp& r) const noexcept { return Fp(*this) *= r; } constexpr Fp operator / (const Fp& r) const noexcept { return Fp(*this) /= r; } constexpr Fp& operator += (const Fp& r) noexcept { val += r.val; if (val >= mod) val -= mod; return *this; } constexpr Fp& operator -= (const Fp& r) noexcept { val -= r.val; if (val < 0) val += mod; return *this; } constexpr Fp& operator *= (const Fp& r) noexcept { val = val * r.val % mod; return *this; } constexpr Fp inv() const noexcept { long long a = val, b = mod, u = 1, v = 0; while (b) { long long t = a / b; a -= t * b; swap(a, b); u -= t * v; swap(u, v); } assert(a == 1); return u; } constexpr Fp& operator /= (const Fp& r) noexcept { return *this *= r.inv(); } constexpr Fp pow(long long n) const noexcept { assert(0 <= n); Fp x = *this, r = 1; while (n) { if (n & 1) r *= x; x *= x; n >>= 1; } return r; } constexpr bool operator == (const Fp& r) const noexcept { return this->val == r.val; } constexpr bool operator != (const Fp& r) const noexcept { return this->val != r.val; } friend constexpr istream& operator >> (istream &is, Fp& x) noexcept { int val; is >> val; x = Fp(val); return is; } friend constexpr ostream& operator << (ostream &os, const Fp& x) noexcept { return os << x.val; } }; //const int mod = 1e9+7; const int mod = 998244353; using mint = Fp; const int MAX_N = 200005; vector g[MAX_N]; mint a[MAX_N]; struct var { mint s; var(mint s_ = 0): s(s_) {} var operator+ (const var& a) const { return {s+a.s}; } var addRoot(int u) const { return {(s+1)*a[u]}; } }; vector dp[MAX_N]; var ans[MAX_N]; var dfs1(int v, int p = -1) { var dpsum; dp[v] = vector (g[v].size()); rep(i, g[v].size()) { int u = g[v][i]; if(u == p) continue; dp[v][i] = dfs1(u, v); dpsum = dpsum + dp[v][i]; } return dpsum.addRoot(v); } void dfs2(int v, const var& dpParent = var(), int p = -1) { int deg = g[v].size(); rep(i, deg) if(g[v][i] == p) dp[v][i] = dpParent; vector dpSum_Left(deg+1), dpSum_Right(deg+1); rep(i,deg) dpSum_Left[i+1] = dpSum_Left[i] + dp[v][i]; rrep(i, deg) dpSum_Right[i] = dpSum_Right[i+1] + dp[v][i]; ans[v] = dpSum_Left[deg].addRoot(v); rep(i,deg) { int u = g[v][i]; if(u == p) continue; var d = dpSum_Left[i] + dpSum_Right[i+1]; dfs2(u, d.addRoot(v), v); } } int main(){ ios_base::sync_with_stdio(false); cin.tie(nullptr); int n; in(n); rep(i,n) in(a[i]); rep(i,n-1){ int u,v; in(u,v); u--; v--; g[u].pb(v); g[v].pb(u); } dfs1(0); dfs2(0); mint res = 0; rep(i,n) res += ans[i].s; rep(i,n) res -= a[i]; out(res/2); }