//Author: Kevin
#include<bits/stdc++.h>
//#pragma GCC optimize("O2")
using namespace std;
#define ll long long
#define ull unsigned ll
#define pb emplace_back
#define mp make_pair
#define ALL(x) (x).begin(),(x).end()
#define rALL(x) (x).rbegin(),(x).rend()
#define srt(x) sort(ALL(x))
#define rev(x) reverse(ALL(x))
#define rsrt(x) sort(rALL(x))
#define sz(x) (int)(x.size())
#define inf 0x3f3f3f3f
#define pii pair<int,int>
#define lb(v,x) (int)(lower_bound(ALL(v),x)-v.begin())
#define ub(v,x) (int)(upper_bound(ALL(v),x)-v.begin())
#define uni(v) v.resize(unique(ALL(v))-v.begin())
#define longer __int128_t
void die(string S){puts(S.c_str());exit(0);}
int n,m;
int a[200200];
vector<int> G[200200];
int in[200200],out[200200],depth[200200],dfn,ind[200200];
int siz[200200],son[200200],tp[200200],f[200200];
void dfs(int u,int fa)
{
	f[u]=fa;
	depth[u]=depth[fa]+1;
	in[u]=++dfn;
	ind[dfn]=u;
	siz[u]=1;
	for(auto v:G[u])
		if(v!=fa)
		{
			dfs(v,u);
			if(siz[v]>siz[son[u]])
				son[u]=v;
			siz[u]+=siz[v];
		}
	out[u]=dfn;
}
void dfs2(int u,int fa)
{
	if(!tp[u]) tp[u]=u;
	if(son[u])
	{
		tp[son[u]]=tp[u];
		dfs2(son[u],u);
	}
	for(auto v:G[u])
		if(v!=fa&&v!=son[u])
			dfs2(v,u);
}
inline int lca(int x,int y)
{
	while(tp[x]!=tp[y])
	{
		if(depth[tp[x]]<depth[tp[y]]) swap(x,y);
		x=f[tp[x]];
	}
	if(depth[x]<=depth[y]) return x;
	return y;
}
vector<ll> dp[200200];
vector<int> pos[200200];
ll Value;
set<int> st;
inline void ins(int x)
{
	Value+=depth[x];
	auto it=st.lower_bound(in[x]);
	if(it!=st.end()&&it!=st.begin())
		Value+=depth[lca(ind[*it],ind[*prev(it)])];
	if(it!=st.end())
		Value-=depth[lca(ind[*it],x)];
	if(it!=st.begin())
		Value-=depth[lca(ind[*prev(it)],x)];
	st.insert(in[x]);
}
inline void del(int x)
{
	Value-=depth[x];
	st.erase(in[x]);
	auto it=st.lower_bound(in[x]);
	if(it!=st.end()&&it!=st.begin())
		Value-=depth[lca(ind[*it],ind[*prev(it)])];
	if(it!=st.end())
		Value+=depth[lca(ind[*it],x)];
	if(it!=st.begin())
		Value+=depth[lca(ind[*prev(it)],x)];
}
void solve(int layer,int l,int r,int L,int R)
{
	if(l>r) return ;
	int mid=(l+r)/2;
	for(int i=max(l,R+1);i<=mid;i++)
		ins(a[i]);
	ll mx=0;
	int mxp=min(R,mid);
	for(int i=min(R,mid);i>=L&&i>=pos[mid][layer-1];i--)
	{
		if(Value+dp[i][layer-1]>mx)
		{
			mx=Value+dp[i][layer-1];
			mxp=i;
		}
		if(i>L)
			ins(a[i]);
	}
	dp[mid][layer]=mx;
	pos[mid][layer]=mxp;
	for(int i=min(R,mid);i>L;i--)
		del(a[i]);
	solve(layer,mid+1,r,mxp,R);
	for(int i=max(l,R+1);i<=mid;i++)
		del(a[i]);
	for(int i=mxp+1;i<=min(R,l-1);i++)
		ins(a[i]);
	solve(layer,l,mid-1,L,mxp);
	for(int i=mxp+1;i<=min(R,l-1);i++)
		del(a[i]);
}
int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		G[u].pb(v);
		G[v].pb(u);
	}
	dfs(1,0);
	dfs2(1,0);
	for(int i=0;i<=n;i++)
		dp[i].resize(m+1);
	for(int i=0;i<=n;i++)
		pos[i].resize(m+1);
	for(int i=1;i<=m;i++)
		solve(i,1,n,0,n);
	cout<<(dp[n][m]-m)*2<<'\n';
	return 0;
}