#include #include using namespace std; using mint = atcoder::modint998244353; int main() { ios::sync_with_stdio(false); cin.tie(0); int n; cin >> n; vector> g(n); vector a(n); for(int i = 1; i < n; i++){ int u, v; cin >> u >> v; g[--u].push_back(--v); g[v].push_back(u); } for(auto &&v : a) cin >> v; mint ans; auto dfs = [&](auto dfs, int v, int p) -> int { mint sml, lrg; int res = 1; for(auto &&u : g[v]){ if(u == p) continue; int sz = dfs(dfs, u, v); res += sz; if(a[u] < a[v]){ ans += sml * sz; sml += sz; }else if(a[u] > a[v]){ ans += lrg * sz; lrg += sz; } } if(p != -1){ if(a[p] < a[v]){ ans += sml * (n - res); }else if(a[p] > a[v]){ ans += lrg * (n - res); } } return res; }; dfs(dfs, 0, -1); cout << ans.val() << '\n'; }