O David vai festejar o seu aniversário e decidiu convidar alguns dos empregados da sua empresa, da qual ele é o diretor executivo (o CEO).
Cada empregado, incluindo o David, é identificado por um número entre 1 e N, um tipo Vi de piadas que sabe contar. Além disso, cada empregado (excepto o David) tem exatamente um supervisor. Como o David é o CEO, ele tem identificador 1 e é direta ou indiretamente chefe de todos (ou seja, todos são seus subordinados).
Na festa de aniversário, existem certas regras que todos (incluindo o David) devem seguir:
Os números de um conjunto são consecutivos se a diferença entre elementos adjacentes é 1 quando o conjunto está ordenado por ordem crescente, como por exemplo (3,1,2) ou (5,1,2,4,3).
O David quer saber quantos conjuntos diferentes de piadas que ele pode ouvir na sua festa, se forem obedecidas as suas regras.
A primeira linha de input contém inteiros N, o número de pessoas da empresa (incluindo o David).
A segunda linha contém N inteiros Vi, os tipos de piada que cada trabalhador conta.
Cada uma das N-1 linhas seguintes contém dois inteiros: A e B, indicando que a pessoa A é um supervisor directo de B (ou seja, que B é um subordinado directo de A).
Uma linha contendo um inteiro, a quantidade de diferentes conjuntos de piadas que obedecem aos requisitos atrás descritos.
São garantidos os seguintes limites em todos os casos de teste que irão ser colocados ao programa:
1 ≤ N ≤ 10 000 | Número total de empregados | |
1 ≤ Vi ≤ 100 | Tipos de piadas | |
1 ≤ A,B ≤ N | Idenficadores das pessoas numa relação directa de supervisão |
Os casos de teste deste problema estão organizados em 2 grupos com restrições adicionais diferentes:
Grupo | Nº de Pontos | Restrições adicionais |
---|---|---|
1 | 50 | 1 ≤ N ≤ 100 |
2 | 50 | --- |
4 2 1 3 4 1 2 1 3 3 4
6
É possível ter os seguintes conjuntos de piadas na festa: {2}, {2,3}, {2,3,4}, {1,2,3,4}, {1,2}, {1,2,3}.
4 3 4 5 6 1 2 1 3 2 4
3
Os únicos conjuntos possíveis são {3}, {3,4} e {3,4,5}. Nota que a pessoa que conta a piada 6 não pode estar na festa porque, nesse caso, o conjunto de piadas {4,6} não é um conjunto de números consecutivos
6 5 3 6 4 2 1 1 2 1 3 1 4 2 5 5 6
10