#include <bits/stdc++.h>
using namespace std;
vector<int> mp[300006];
int node[300006];
struct color {
  int r, b, o;
} ans[300006];
int sum;
color dfs(int u, int fa) {
  if (node[u] == 1) ans[u].r = 1, ans[u].b = ans[u].o = 0;
  if (node[u] == 2) ans[u].b = 1, ans[u].r = ans[u].o = 0;
  if (node[u] == 0) ans[u].o = 1, ans[u].b = ans[u].r = 0;
  for (int i = 0; i < mp[u].size(); i++) {
    int v = mp[u][i];
    if (v == fa) continue;
    color res;
    res = dfs(v, u);
    ans[u].r += res.r;
    ans[u].b += res.b;
    ans[u].o += res.o;
  }
  return ans[u];
}
void dfs2(int u, int fa) {
  for (int i = 0; i < mp[u].size(); i++) {
    int v = mp[u][i];
    if (v == fa) continue;
    if (!((ans[1].r - ans[v].r > 0 && ans[1].b - ans[v].b > 0) ||
          (ans[v].r > 0 && ans[v].b > 0))) {
      sum++;
    }
    dfs2(v, u);
  }
}
int main() {
  cin.tie(0);
  ios::sync_with_stdio(0);
  int n;
  while (cin >> n) {
    for (int i = 1; i <= n; i++) {
      cin >> node[i];
      mp[i].clear();
    }
    int u, v;
    for (int i = 1; i < n; i++) {
      cin >> u >> v;
      mp[u].push_back(v);
      mp[v].push_back(u);
    }
    dfs(1, -1);
    sum = 0;
    dfs2(1, -1);
    cout << sum << endl;
  }
  return 0;
}
