1
#include<bits/stdc++.h>
2

3
using namespace std;
4

5
//----------------------------------------
6
const int N = 1e6 + 10;
7
int n;
8
bool color[N];
9
int fa[N], depth[N];
10
int head[N], nxt[N << 1], to[N << 1], cnt;
11
int vis[N];
12
long long white_dis_sum;
13
int vis_of_p[N];
14
void add(int x, int y)
15
{
16
    to[++cnt] = y;
17
    nxt[cnt] = head[x];
18
    head[x] = cnt;
19
}
20

21
void dfs(int u, int anc, int first_white)
22
{
23
    depth[u] = depth[anc] + 1;
24
    fa[u] = first_white;
25
    if(color[u] == 0) white_dis_sum += depth[u];
26
    for(int i = head[u]; i; i = nxt[i])
27
    {
28
        int v = to[i];
29
        if(v == anc) continue;
30
        if(color[u]/*black*/)
31
        {
32
            dfs(v, u, first_white);
33
        }
34
        else/*white*/
35
        {
36
            dfs(v, u, u);
37
        }
38
    }
39
}
40

41
int findfa_with_vis_equals_to_0(int u)
42
{
43
//  cout << "qaq" << u << endl;
44
    if(u == -1 || u == 0x3f3f3f3f) return 0x3f3f3f3f;
45
    if(vis[u] == 0) return u;
46
    return fa[u] = findfa_with_vis_equals_to_0(fa[u]);
47
}
48

49
int main()
50
{
51
    // for(int i = 1;i <= n;i++)
52
    // {
53
    //     fa[i] = i;
54
    // }
55
//    clock_t c1 = clock();
56
// #ifdef LOCAL
57
    freopen("wbtree.in", "r", stdin);
58
    freopen("wbtree.out", "w", stdout);
59
// #endif
60
//----------------------------------------
61
    ios::sync_with_stdio(0), cin.tie(0);
62
    cin >> n;
63
    for(int i = 1;i <= n;i++)
64
        cin >> color[i];
65
    int u, v;
66
    for(int i = 1;i <= n - 1;i++)
67
    {
68
        cin >> u >> v;
69
        add(u, v), add(v, u);
70
    }
71
    //找不到就是-1
72
    dfs(1, 0, -1);
73

74
    queue <int> qwq;
75
    qwq.push(1);
76
    long long ans = 0;
77
//    for(int i = 1;i <= n;i++)
78
//    {
79
//      cout << i << ": " << fa[i] << endl;
80
//  }
81
    while(qwq.size())
82
    {
83
        int t = qwq.front();
84
//      cout << "qwq"<<t << endl;
85
    vis_of_p[t] = 1;
86
        for(int i = head[t]; i; i = nxt[i])
87
        {
88
            int v = to[i];
89
            if(vis_of_p[v]) continue;
90
            qwq.push(v);
91
        }
92
        qwq.pop();
93
        if(color[t] == 0) continue;
94
        int deepest_white = findfa_with_vis_equals_to_0(fa[t]);
95
//        cout << t << "的deepest_white: " << deepest_white << endl;
96
        if(deepest_white == 0x3f3f3f3f) continue;
97
        vis[deepest_white] = 1;
98
        ans += depth[t];
99
    }
100
    cout <<  ans - white_dis_sum;
101
//----------------------------------------
102
//end:
103
//    cerr << "Time Used:" << clock() - c1 << "ms" << endl;
104
    return 0;
105
}
106

107
/*
108
 * ┌───┐   ┌───┬───┬───┬───┐ ┌───┬───┬───┬───┐ ┌───┬───┬───┬───┐ ┌───┬───┬───┐
109
 * │Esc│   │ F1│ F2│ F3│ F4│ │ F5│ F6│ F7│ F8│ │ F9│F10│F11│F12│ │P/S│S L│P/B│  ┌┐    ┌┐    ┌┐
110
 * └───┘   └───┴───┴───┴───┘ └───┴───┴───┴───┘ └───┴───┴───┴───┘ └───┴───┴───┘  └┘    └┘    └┘
111
 * ┌───┬───┬───┬───┬───┬───┬───┬───┬───┬───┬───┬───┬───┬───────┐ ┌───┬───┬───┐ ┌───┬───┬───┬───┐
112
 * │~ `│! 1│@ 2│# 3│$ 4│% 5│^ 6│& 7│* 8│( 9│) 0│_ -│+ =│ BacSp │ │Ins│Hom│PUp│ │Num│ / │ * │ - │
113
 * ├───┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─────┤ ├───┼───┼───┤ ├───┼───┼───┼───┤
114
 * │ Tab │ Q │ W │ E │ R │ T │ Y │ U │ I │ O │ P │{ [│} ]│ | \ │ │Del│End│PDn│ │ 7 │ 8 │ 9 │   │
115
 * ├─────┴┬──┴┬──┴┬──┴┬──┴┬──┴┬──┴┬──┴┬──┴┬──┴┬──┴┬──┴┬──┴─────┤ └───┴───┴───┘ ├───┼───┼───┤ + │
116
 * │ Caps │ A │ S │ D │ F │ G │ H │ J │ K │ L │: ;│" '│ Enter  │               │ 4 │ 5 │ 6 │   │
117
 * ├──────┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴─┬─┴────────┤     ┌───┐     ├───┼───┼───┼───┤
118
 * │ Shift  │ Z │ X │ C │ V │ B │ N │ M │< ,│> .│? /│  Shift   │     │ ↑ │     │ 1 │ 2 │ 3 │   │
119
 * ├─────┬──┴─┬─┴──┬┴───┴───┴───┴───┴───┴──┬┴───┼───┴┬────┬────┤ ┌───┼───┼───┐ ├───┴───┼───┤ E││
120
 * │ Ctrl│ Win│ Alt│         Space         │ Alt│ Win│Menu│Ctrl│ │ ← │ ↓ │ → │ │   0   │ . │←─┘│
121
 * └─────┴────┴────┴───────────────────────┴────┴────┴────┴────┘ └───┴───┴───┘ └───────┴───┴───┘
122
 */