#line 2 "/Users/noya2/Desktop/Noya2_library/template/template.hpp" using namespace std; #include #line 1 "/Users/noya2/Desktop/Noya2_library/template/inout_old.hpp" namespace noya2 { template 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...); } template void out(const vector> &vv){ int s = (int)vv.size(); for (int i = 0; i < s; i++) out(vv[i]); } struct IoSetup { IoSetup(){ cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(15); cerr << fixed << setprecision(7); } } iosetup_noya2; } // namespace noya2 #line 1 "/Users/noya2/Desktop/Noya2_library/template/const.hpp" namespace noya2{ const int iinf = 1'000'000'007; const long long linf = 2'000'000'000'000'000'000LL; const long long mod998 = 998244353; const long long mod107 = 1000000007; const long double pi = 3.14159265358979323; const vector dx = {0,1,0,-1,1,1,-1,-1}; const vector dy = {1,0,-1,0,1,-1,-1,1}; const string ALP = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; const string alp = "abcdefghijklmnopqrstuvwxyz"; const string NUM = "0123456789"; void yes(){ cout << "Yes\n"; } void no(){ cout << "No\n"; } void YES(){ cout << "YES\n"; } void NO(){ cout << "NO\n"; } void yn(bool t){ t ? yes() : no(); } void YN(bool t){ t ? YES() : NO(); } } // namespace noya2 #line 2 "/Users/noya2/Desktop/Noya2_library/template/utils.hpp" #line 6 "/Users/noya2/Desktop/Noya2_library/template/utils.hpp" namespace noya2{ unsigned long long inner_binary_gcd(unsigned long long a, unsigned long long b){ if (a == 0 || b == 0) return a + b; int n = __builtin_ctzll(a); a >>= n; int m = __builtin_ctzll(b); b >>= m; while (a != b) { int mm = __builtin_ctzll(a - b); bool f = a > b; unsigned long long c = f ? a : b; b = f ? b : a; a = (c - b) >> mm; } return a << std::min(n, m); } template T gcd_fast(T a, T b){ return static_cast(inner_binary_gcd(std::abs(a),std::abs(b))); } long long sqrt_fast(long long n) { if (n <= 0) return 0; long long x = sqrt(n); while ((x + 1) * (x + 1) <= n) x++; while (x * x > n) x--; return x; } template T floor_div(const T n, const T d) { assert(d != 0); return n / d - static_cast((n ^ d) < 0 && n % d != 0); } template T ceil_div(const T n, const T d) { assert(d != 0); return n / d + static_cast((n ^ d) >= 0 && n % d != 0); } template void uniq(std::vector &v){ std::sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); } template inline bool chmin(T &x, U y) { return (y < x) ? (x = y, true) : false; } template inline bool chmax(T &x, U y) { return (x < y) ? (x = y, true) : false; } template inline bool range(T l, T x, T r){ return l <= x && x < r; } } // namespace noya2 #line 8 "/Users/noya2/Desktop/Noya2_library/template/template.hpp" #define rep(i,n) for (int i = 0; i < (int)(n); i++) #define repp(i,m,n) for (int i = (m); i < (int)(n); i++) #define reb(i,n) for (int i = (int)(n-1); i >= 0; i--) #define all(v) (v).begin(),(v).end() using ll = long long; using ld = long double; using uint = unsigned int; using ull = unsigned long long; using pii = pair; using pll = pair; using pil = pair; using pli = pair; namespace noya2{ /* ~ (. _________ . /) */ } using namespace noya2; #line 2 "c.cpp" #line 2 "/Users/noya2/Desktop/Noya2_library/tree/rerootingdp.hpp" #line 5 "/Users/noya2/Desktop/Noya2_library/tree/rerootingdp.hpp" #include namespace noya2 { // g[from] contains outgoing edges (to, edgeid(from, to)) // (E, op, e) is commutative monoid // ~edgeid(from, to) == edgeid(to, from) // return calculator of dp(r, v) template auto rerootingdp(auto op, E e, auto put_edge, auto put_vertex, const std::vector>> &g, int root = 0){ struct dp { // dp(r, v) : root is r, dp of subtree v // ans[v] = dp(v, v) // from[v] = dp(v, par(v)) // to[v] = dp(par(v), v) // from[root] and to[root] is undefined std::vector ans, from, to; std::vector down, up; std::vector> childs; bool is_in_subtree(int r, int v){ return down[r] < down[v] && up[v] <= up[r]; } V operator()(int r, int v){ if (r == v) return ans[v]; if (!is_in_subtree(v, r)) return to[v]; int le = 0, ri = childs[v].size(); while (ri - le > 1){ int md = (le + ri) / 2; if (down[childs[v][md]] <= down[r]){ le = md; } else { ri = md; } } return from[childs[v][le]]; } }; int n = g.size(); std::vector from(n, e), to(n, e); std::vector dp_to(n); std::vector> childs(n); std::vector down(n), up(n); int t = 0; auto dfs = [&](auto sfs, int v, int f) -> void { down[v] = t++; childs[v].reserve(g[v].size()); for (auto [c, eid] : g[v]){ if (c == f) continue; childs[v].emplace_back(c); sfs(sfs, c, v); dp_to[c] = put_vertex(to[c], c); to[c] = put_edge(dp_to[c], eid); to[v] = op(to[v], to[c]); } up[v] = t; }; dfs(dfs, root, -1); std::vector dp_ans(n), dp_from(n); std::vector inner(n); auto bfs = [&](auto sfs, int v, int f) -> void { int sz = g[v].size(); inner[sz] = e; int i = sz-1; for (auto [c, eid] : g[v] | std::views::reverse){ if (c == f) continue; inner[i] = op(inner[i+1], to[c]); i--; } dp_ans[v] = put_vertex(op(inner[++i], from[v]), v); E rui = e; for (auto [c, eid] : g[v]){ if (c == f) continue; dp_from[c] = put_vertex(op(op(rui, inner[++i]), from[v]), v); from[c] = put_edge(dp_from[c], ~eid); rui = op(rui, to[c]); } for (auto [c, eid] : g[v]){ if (c == f) continue; sfs(sfs, c, v); } }; bfs(bfs, root, -1); return dp{dp_ans, dp_from, dp_to, down, up, childs}; } } // namespace noya2 #line 2 "/Users/noya2/Desktop/Noya2_library/utility/modint.hpp" #line 4 "/Users/noya2/Desktop/Noya2_library/utility/modint.hpp" #line 2 "/Users/noya2/Desktop/Noya2_library/math/prime.hpp" #line 4 "/Users/noya2/Desktop/Noya2_library/math/prime.hpp" namespace noya2 { constexpr long long safe_mod(long long x, long long m) { x %= m; if (x < 0) x += m; return x; } constexpr long long pow_mod_constexpr(long long x, long long n, int m) { if (m == 1) return 0; unsigned int _m = (unsigned int)(m); unsigned long long r = 1; unsigned long long y = safe_mod(x, m); while (n) { if (n & 1) r = (r * y) % _m; y = (y * y) % _m; n >>= 1; } return r; } constexpr bool is_prime_constexpr(int n) { if (n <= 1) return false; if (n == 2 || n == 7 || n == 61) return true; if (n % 2 == 0) return false; long long d = n - 1; while (d % 2 == 0) d /= 2; constexpr long long bases[3] = {2, 7, 61}; for (long long a : bases) { long long t = d; long long y = pow_mod_constexpr(a, t, n); while (t != n - 1 && y != 1 && y != n - 1) { y = y * y % n; t <<= 1; } if (y != n - 1 && t % 2 == 0) { return false; } } return true; } template constexpr bool is_prime_flag = is_prime_constexpr(n); constexpr std::pair inv_gcd(long long a, long long b) { a = safe_mod(a, b); if (a == 0) return {b, 0}; long long s = b, t = a; long long m0 = 0, m1 = 1; while (t) { long long u = s / t; s -= t * u; m0 -= m1 * u; auto tmp = s; s = t; t = tmp; tmp = m0; m0 = m1; m1 = tmp; } if (m0 < 0) m0 += b / s; return {s, m0}; } constexpr int primitive_root_constexpr(int m) { if (m == 2) return 1; if (m == 167772161) return 3; if (m == 469762049) return 3; if (m == 754974721) return 11; if (m == 998244353) return 3; int divs[20] = {}; divs[0] = 2; int cnt = 1; int x = (m - 1) / 2; while (x % 2 == 0) x /= 2; for (int i = 3; (long long)(i)*i <= x; i += 2) { if (x % i == 0) { divs[cnt++] = i; while (x % i == 0) { x /= i; } } } if (x > 1) { divs[cnt++] = x; } for (int g = 2;; g++) { bool ok = true; for (int i = 0; i < cnt; i++) { if (pow_mod_constexpr(g, (m - 1) / divs[i], m) == 1) { ok = false; break; } } if (ok) return g; } } template constexpr int primitive_root_flag = primitive_root_constexpr(m); // constexpr long long primitive_root_constexpr(long long m){ // if (m == (1LL << 47) - (1LL << 24) + 1) return 3; // return primitive_root_constexpr(static_cast(m)); // } } // namespace noya2 #line 6 "/Users/noya2/Desktop/Noya2_library/utility/modint.hpp" namespace noya2{ struct barrett { unsigned int _m; unsigned long long im; explicit barrett(unsigned int m) : _m(m), im((unsigned long long)(-1) / m + 1) {} unsigned int umod() const { return _m; } unsigned int mul(unsigned int a, unsigned int b) const { unsigned long long z = a; z *= b; unsigned long long x = (unsigned long long)((__uint128_t(z) * im) >> 64); unsigned int v = (unsigned int)(z - x * _m); if (_m <= v) v += _m; return v; } }; template struct static_modint { using mint = static_modint; public: static constexpr int mod() { return m; } static mint raw(int v) { mint x; x._v = v; return x; } constexpr static_modint() : _v(0) {} template constexpr static_modint(T v){ long long x = (long long)(v % (long long)(umod())); if (x < 0) x += umod(); _v = (unsigned int)(x); } template constexpr static_modint(T v){ _v = (unsigned int)(v % umod()); } constexpr unsigned int val() const { return _v; } mint& operator++() { _v++; if (_v == umod()) _v = 0; return *this; } mint& operator--() { if (_v == 0) _v = umod(); _v--; return *this; } mint operator++(int) { mint result = *this; ++*this; return result; } mint operator--(int) { mint result = *this; --*this; return result; } constexpr mint& operator+=(const mint& rhs) { _v += rhs._v; if (_v >= umod()) _v -= umod(); return *this; } constexpr mint& operator-=(const mint& rhs) { _v -= rhs._v; if (_v >= umod()) _v += umod(); return *this; } constexpr mint& operator*=(const mint& rhs) { unsigned long long z = _v; z *= rhs._v; _v = (uint)(z % umod()); return *this; } constexpr mint& operator/=(const mint& rhs) { return *this = *this * rhs.inv(); } constexpr mint operator+() const { return *this; } constexpr mint operator-() const { return mint() - *this; } constexpr mint pow(long long n) const { assert(0 <= n); mint x = *this, r = 1; while (n) { if (n & 1) r *= x; x *= x; n >>= 1; } return r; } constexpr mint inv() const { if (prime) { assert(_v); return pow(umod() - 2); } else { auto eg = inv_gcd(_v, m); assert(eg.first == 1); return eg.second; } } friend constexpr mint operator+(const mint& lhs, const mint& rhs) { return mint(lhs) += rhs; } friend constexpr mint operator-(const mint& lhs, const mint& rhs) { return mint(lhs) -= rhs; } friend constexpr mint operator*(const mint& lhs, const mint& rhs) { return mint(lhs) *= rhs; } friend constexpr mint operator/(const mint& lhs, const mint& rhs) { return mint(lhs) /= rhs; } friend constexpr bool operator==(const mint& lhs, const mint& rhs) { return lhs._v == rhs._v; } friend constexpr bool operator!=(const mint& lhs, const mint& rhs) { return lhs._v != rhs._v; } friend std::ostream &operator<<(std::ostream &os, const mint& p) { return os << p.val(); } friend std::istream &operator>>(std::istream &is, mint &a) { long long t; is >> t; a = mint(t); return (is); } private: unsigned int _v; static constexpr unsigned int umod() { return m; } static constexpr bool prime = is_prime_flag; }; template struct dynamic_modint { using mint = dynamic_modint; public: static int mod() { return (int)(bt.umod()); } static void set_mod(int m) { assert(1 <= m); bt = barrett(m); } static mint raw(int v) { mint x; x._v = v; return x; } dynamic_modint() : _v(0) {} template dynamic_modint(T v){ long long x = (long long)(v % (long long)(umod())); if (x < 0) x += umod(); _v = (unsigned int)(x); } template dynamic_modint(T v){ _v = (unsigned int)(v % umod()); } uint val() const { return _v; } mint& operator++() { _v++; if (_v == umod()) _v = 0; return *this; } mint& operator--() { if (_v == 0) _v = umod(); _v--; return *this; } mint operator++(int) { mint result = *this; ++*this; return result; } mint operator--(int) { mint result = *this; --*this; return result; } mint& operator+=(const mint& rhs) { _v += rhs._v; if (_v >= umod()) _v -= umod(); return *this; } mint& operator-=(const mint& rhs) { _v += mod() - rhs._v; if (_v >= umod()) _v -= umod(); return *this; } mint& operator*=(const mint& rhs) { _v = bt.mul(_v, rhs._v); return *this; } mint& operator/=(const mint& rhs) { return *this = *this * rhs.inv(); } mint operator+() const { return *this; } mint operator-() const { return mint() - *this; } mint pow(long long n) const { assert(0 <= n); mint x = *this, r = 1; while (n) { if (n & 1) r *= x; x *= x; n >>= 1; } return r; } mint inv() const { auto eg = noya2::inv_gcd(_v, mod()); assert(eg.first == 1); return eg.second; } friend mint operator+(const mint& lhs, const mint& rhs) { return mint(lhs) += rhs; } friend mint operator-(const mint& lhs, const mint& rhs) { return mint(lhs) -= rhs; } friend mint operator*(const mint& lhs, const mint& rhs) { return mint(lhs) *= rhs; } friend mint operator/(const mint& lhs, const mint& rhs) { return mint(lhs) /= rhs; } friend bool operator==(const mint& lhs, const mint& rhs) { return lhs._v == rhs._v; } friend bool operator!=(const mint& lhs, const mint& rhs) { return lhs._v != rhs._v; } friend std::ostream &operator<<(std::ostream &os, const mint& p) { return os << p.val(); } friend std::istream &operator>>(std::istream &is, mint &a) { long long t; is >> t; a = mint(t); return (is); } private: unsigned int _v; static barrett bt; static unsigned int umod() { return bt.umod(); } }; template noya2::barrett dynamic_modint::bt(998244353); using modint998244353 = static_modint<998244353>; using modint1000000007 = static_modint<1000000007>; using modint = dynamic_modint<-1>; template concept Modint = requires (T &a){ T::mod(); a.inv(); a.val(); a.pow(declval()); }; } // namespace noya2 #line 5 "c.cpp" using mint = modint998244353; void solve(){ int n; in(n); vector a(n); in(a); vector> g(n); rep(i,n-1){ int u, v; in(u,v); u--, v--; g[u].emplace_back(v,i); g[v].emplace_back(u,~i); } auto op = [&](mint l, mint r){ return l + r; }; auto pute = [&](mint v, int){ return v; }; auto putv = [&](mint e, int v){ return e * a[v] + a[v]; }; auto dp = rerootingdp(op,mint(0),pute,putv,g); mint ans = 0; rep(i,n){ ans += dp(i,i); ans -= a[i]; } ans /= 2; out(ans); } int main(){ int t = 1; //in(t); while (t--) { solve(); } }