Đường đẹp

Có N thà nh phố trong mộ t đấ t nướ c đượ c kết nố i bằ ng đườ ng hai chiều. Mộ t số thà nh
phố là thà nh phố loạ i 1, mộ t số là loạ i 2 và mộ t số chưa đượ c xếp loạ i. Đấ t nướ c đượ c
đả m bả o rằ ng có chứ a ít nhấ t mộ t thà nh phố loạ i 1, mộ t thà nh phố loạ i 2. Bạ n chọ n mộ t
con đườ ng và loạ i bỏ nó ra khỏ i hệ thố ng đườ ng đi khiến đấ t nướ c bị chia thà nh hai
phầ n. Đó sẽ là mộ t con đườ ng đẹp nếu nó chia đấ t nướ c thà nh hai phầ n mà kết quả mỗ i
phầ n khô ng chứ a cá c thà nh phố củ a cả hai loạ i 1 và 2.

Bạ n hã y tính số lượ ng nhữ ng con đườ ng đẹp.

INPUT: Và o từ file vă n bả n BROAD.INP:

- Dò ng đầ u tiên chứ a số nguyên N (3 ≤ N ≤ 3.10 5), số thà nh phố . Cá c thà nh phố

đượ c dá n nhã n vớ i số lượ ng từ 1 đến N.

- Dò ng thứ hai chứ a N số nguyên a1, a2,…,an (0 ≤ ai ≤ 2) là loạ i củ a cá c thà nh phố

vớ i:

 ai = 1 có nghĩa là thà nh phố loạ i 1

 ai = 2 có nghĩa là thà nh phố loạ i 2
 ai = 0 có nghĩa là thà nh phố chưa đượ c xếp loạ i
-Dò ng thứ i củ a N-1 dò ng tiếp theo chứ a hai số nguyên v i, ui (1 ≤ vi, ui ≤ N)– các cạ nh củ a
câ y.

  OUTPUT: Đưa ra file vă n bản BROAD.OUT Chỉ có mộ t số nguyên duy nhấ t là số lượ ng
đườ ng đẹp

BROAD.INP BROAD.OUT
5 1
20012
12
23
24
25

Sol:
Tính toá n số lượ ng đỉnh củ a đồ thị đượ c đá nh dấu 1 hoặc 2 trong từ ng câ y con và kiểm tra điều
kiện thích hợ p trong khi duyệt đồ thị.