#include <bits/stdc++.h>
using namespace std;
vector<int> sused[300010];
bool obidjen[300010];
int predak[300010];
int broj_plavih[300010];
int broj_crvenih[300010];
int boja[300010];
void DFS(int v) {
  if (obidjen[v]) return;
  obidjen[v] = true;
  for (int i = 0; i < sused[v].size(); i++) {
    if (!obidjen[sused[v][i]]) {
      predak[sused[v][i]] = v;
      DFS(sused[v][i]);
    }
  }
  broj_crvenih[v] = 0;
  broj_plavih[v] = 0;
  if (boja[v] == 1) broj_plavih[v] = 1;
  if (boja[v] == 2) broj_crvenih[v] = 1;
  for (int i = 0; i < sused[v].size(); i++) {
    if (predak[sused[v][i]] == v) {
      broj_crvenih[v] += broj_crvenih[sused[v][i]];
      broj_plavih[v] += broj_plavih[sused[v][i]];
    }
  }
}
int main() {
  int n;
  scanf("%d", &n);
  for (int i = 1; i <= n; i++) scanf("%d", &boja[i]);
  int v, u;
  for (int i = 0; i < n - 1; i++) {
    scanf("%d%d", &v, &u);
    sused[v].push_back(u);
    sused[u].push_back(v);
  }
  for (int i = 1; i <= n; i++) obidjen[i] = false;
  DFS(1);
  int rez = 0;
  for (int i = 2; i <= n; i++) {
    if (broj_crvenih[i] == broj_crvenih[1] && broj_plavih[i] == 0) rez++;
    if (broj_plavih[i] == broj_plavih[1] && broj_crvenih[i] == 0) rez++;
  }
  printf("%d\n", rez);
  return 0;
}
