记录编号	584792	评测结果	AAAAAAAAAA
题目名称	树权	最终得分	100
用户昵称	┭┮﹏┭┮	是否通过	通过
代码语言	C++	运行时间	0.906 s
提交时间	2023-11-15 17:26:49	内存使用	24.04 MiB

#include <bits/stdc++.h> 
using namespace std;
typedef long long ll;
const int N = 1e6+10;
int n,m;
ll ans;
struct side{
    int ver,nx;
}e[N<<1];
int hd[N],tot;
void add(int x,int y){
    tot++;
    e[tot].ver = y,e[tot].nx = hd[x],hd[x] = tot;
}
struct made{
    int l,r;
}a[N];
int s[N<<1];
int cnt;
void dfs(int x,int fa){
    for(int i = hd[x];i;i = e[i].nx){
        int y = e[i].ver;
        if(y == fa)continue;
        dfs(y,x);
    }
    for(int i = hd[x];i;i = e[i].nx){
        int y = e[i].ver;
        if(y == fa)continue;
        s[++cnt] = a[y].l,s[++cnt] = a[y].r;
    }
    if(!hd[x] || !cnt)return;
    sort(s+1,s+1+cnt);
    a[x].l = s[cnt>>1],a[x].r = s[(cnt>>1)+1];
    cnt = 0;
    for(int i = hd[x];i;i = e[i].nx){
        int y = e[i].ver;
        if(y == fa)continue;
        if(a[y].r < a[x].l)ans += 1ll * abs(a[y].r - a[x].l);
        if(a[y].l > a[x].l)ans += 1ll * abs(a[y].l - a[x].l);
    }
}
int main(){
    freopen("starria.in","r",stdin);
    freopen("starria.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i = 1;i < n;i++){
        int x,y;scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    for(int i = 1;i <= m;i++){
        int x;scanf("%d",&x);
        a[i].l = a[i].r = x;
    }
    if(n == 2)return !printf("%lld\n",abs(a[1].l-a[2].l));
    dfs(n,0);
    printf("%lld\n",ans);
    
    return 0;
    
}