#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 rep(i,n) for(int i=0; i<(n); i++) #define rrep(i,n) for(int i=(n)-1; i>=0; i--) #define FOR(i,a,b) for(int i=(a); i<(b); i++) #define RFOR(i,a,b) for(int i=(b-1); i>=(a); i--) #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; 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...); } int main(){ ios_base::sync_with_stdio(false); cin.tie(nullptr); int n; in(n); vector> g(n); rep(i,n-1){ int u,v; in(u,v); u--; v--; g[u].pb(v); g[v].pb(u); } vector a(n); in(a); ll ans = 0; auto dfs = [&] (auto dfs, int u, int p) -> int{ int su = 1; ll small = 0, big = 0; ll small2 = 0, big2 = 0; for(int v : g[u]){ if(v == p) continue; int sv = dfs(dfs, v, u); su += sv; if(a[u] > a[v]){ small += sv; small2 += (ll)sv*sv; } else if(a[u] < a[v]){ big += sv; big2 += (ll)sv*sv; } } if(p != -1){ if(a[u] > a[p]){ small += n-su; small2 += (ll)(n-su)*(n-su); } else if(a[u] < a[p]){ big += n-su; big2 += (ll)(n-su)*(n-su); } } ans += small*small - small2; ans += big*big - big2; return su; }; dfs(dfs, 0, -1); out((ans/2) % 998244353); }