#abc163f. [abc163_f]path pass i
[abc163_f]path pass i
問題文
から までの番号がつけられた 個の頂点を持つ木があります。この木の 番目の辺は頂点 と を結んでいます。 また、各頂点には色が塗られており、 頂点 に塗られている色は です。ここで、各頂点に塗られている色は 以上 以下の整数で表されており、同じ整数は同じ色に、異なる整数は異なる色に対応します。
に対して、以下の問題を解いてください。
- 色 が塗られている頂点を一度以上通るような単純パスの数を求めよ
補足: 頂点 から頂点 へ向かう単純パスと、頂点 から頂点 へ向かう単純パスは区別しません。
制約
- 与えられるグラフは木である。
- 入力はすべて整数である。
入力
入力は以下の形式で標準入力から与えられる。
出力
に対する問題の答えを、順番に改行区切りで出力せよ。
入力例 1
3
1 2 1
1 2
2 3
出力例 1
5
4
0
頂点 と頂点 を結ぶ単純パスを、 と表します。
色 が塗られている頂点を一度以上通る単純パスは、
の つあります。
色 が塗られている頂点を一度以上通る単純パスは、
の つあります。
色 が塗られている頂点を一度以上通る単純パスはありません。
入力例 2
1
1
出力例 2
1
入力例 3
2
1 2
1 2
出力例 3
2
2
入力例 4
5
1 2 3 4 5
1 2
2 3
3 4
3 5
出力例 4
5
8
10
5
5
入力例 5
8
2 7 2 5 4 1 7 5
3 1
1 2
2 7
4 5
5 6
6 8
7 8
出力例 5
18
15
0
14
23
0
23
0