#include<bits/stdc++.h>
#define ll long long
#define il inline
#define pn putchar('\n')
#define maxint 2147483647
#define min(x,y) (x<y?x:y)
#define max(x,y) (x>y?x:y)
#define mod 998244353
#define maxn 500005
#define int ll
using namespace std;
struct Gragh
{
    int v,nex;
}e[maxn<<1];
int n,top,tt,ans;
int a[maxn],c[maxn],si[maxn],sum[maxn];
il void build_gra(int u,int v)
{
    e[++top]=(Gragh){v,c[u]},c[u]=top;
}
il void upd(int &x,int y)
{
	x+=y;
	if(x>=mod)x-=mod;
	if(x<0)x+=mod;
}
il void dfs(int pos,int fa)
{
	si[pos]=1;
	for(int i=c[pos];i;i=e[i].nex)
	{
		int v=e[i].v;
		if(v==fa)continue;
		dfs(v,pos);
		si[pos]+=si[v];
		upd(ans,sum[v]*(n-si[v])%mod);
		upd(ans,sum[v]*sum[pos]%mod);
		upd(sum[pos],sum[v]);
	}
	upd(sum[pos],si[pos]);
}
signed main()
{
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%lld%lld",&u,&v);
		build_gra(u,v),build_gra(v,u);
	}
	dfs(1,0);
	printf("%lld",ans);
    return 0;
}