#include<bits/stdc++.h>

#define ll long long
#define mk make_pair
#define fi first
#define se second

using namespace std;

inline int read(){
	int x=0,f=1;char c=getchar();
	for(;(c<'0'||c>'9');c=getchar()){if(c=='-')f=-1;}
	for(;(c>='0'&&c<='9');c=getchar())x=x*10+(c&15);
	return x*f;
}

const int mod=998244353;
int ksm(int x,ll y,int p=mod){
	int ans=1;y%=(p-1);
	for(int i=y;i;i>>=1,x=1ll*x*x%p)if(i&1)ans=1ll*ans*x%p;
	return ans%p;
}
int inv(int x,int p=mod){return ksm(x,p-2,p)%p;}
mt19937 rnd(time(0));
int randint(int l,int r){return rnd()%(r-l+1)+l;}
void add(int &x,int v){x+=v;if(x>=mod)x-=mod;}
void Mod(int &x){if(x>=mod)x-=mod;}
int cmod(int x){if(x>=mod)x-=mod;return x;}

template<typename T>void cmax(T &x,T v){x=max(x,v);}
template<typename T>void cmin(T &x,T v){x=min(x,v);}

void solve(){
	int n=read(),k=read();
	vector<vector<int> >adj(n+1),G(n+1);
	vector<int>a(n+1);
	
	vector<pair<int,int> >edges(n-1);
	for(int i=1;i<=n-1;i++){
		int u=read(),v=read();edges[i-1]=mk(u,v);
		adj[u].emplace_back(v),adj[v].emplace_back(u);
	}
	vector<vector<int> >vec(n+1);
	for(int i=1;i<=n;i++)vec[a[i]=read()].emplace_back(i);
	
	vector<int>In(n+1),dep(n+1),root(n+1),fa(n+1);int tot=0;
	function<void(int)>dfs1=[&](int u){
		if(a[u]!=a[fa[u]])In[u]=++tot,root[a[u]]=u;
		else In[u]=In[fa[u]];
		dep[u]=dep[fa[u]]+1;
		for(int v:adj[u])if(v!=fa[u])fa[v]=u,dfs1(v);
	};
	dfs1(1);
	
	for(int i=1;i<=k;i++){
		if(vec[i].size()==0)return puts("0"),void();
		for(int j=1;j<vec[i].size();j++)if(In[vec[i][j]]!=In[vec[i][j-1]])return puts("0"),void();
	}
	for(auto [u,v]:edges){
		if(a[u]!=a[v])G[a[u]].emplace_back(a[v]),G[a[v]].emplace_back(a[u]);
	}
	
	vector<vector<int> >dis(n+1,vector<int>(n+1));
	for(int i=1;i<=n;i++){
		fill(dis[i].begin(),dis[i].end(),-1);
		queue<int>q;q.push(i),dis[i][i]=0;
		while(q.size()){
			int x=q.front();q.pop();
			for(int y:adj[x])if(dis[i][y]==-1)dis[i][y]=dis[i][x]+1,q.push(y);
		}
	}
	
	vector<vector<int> >f(k+1,vector<int>(n+1));
	function<void(int,int)>DP=[&](int u,int fr){
		for(int v:G[u])if(v!=fr)DP(v,u);
		for(int j:vec[u]){
			f[u][j]=1;
			for(int v:G[u])if(v!=fr){
				int sum=0;
				for(int k:vec[v]){
					int dv=dis[k][root[v]],du=dis[fa[root[v]]][j];
					if(dv==du)add(sum,f[v][k]);
					else if(dv==du+1&&u>v)add(sum,f[v][k]);
					else if(du==dv+1&&u<v)add(sum,f[v][k]);
				}
				f[u][j]=1ll*f[u][j]*sum%mod;
			}
		}
	};
	
	DP(a[1],0);
	int ans=0;for(int i=1;i<=n;i++)add(ans,f[a[1]][i]);
	cout<<ans<<endl;
}

signed main(void){

#ifndef ONLINE_JUDGE
	freopen("in.in","r",stdin);
#endif

	int tt=read();while(tt--)solve();

	return 0;
}