Số màu phân biệt trong cây con
Đề bài
Mô tả
Cho một cây gồm đỉnh, gốc tại đỉnh 1. Mỗi đỉnh có một màu sắc.
Với mỗi đỉnh, hãy tính số lượng màu sắc phân biệt trong cây con gốc tại đỉnh đó.
Dữ liệu vào
Dòng đầu chứa số nguyên .
Dòng thứ hai chứa số nguyên — màu sắc của từng đỉnh.
- dòng tiếp theo, mỗi dòng chứa hai số nguyên và mô tả một cạnh của cây.
Dữ liệu ra
In ra số nguyên, số thứ là số màu phân biệt trong cây con gốc tại đỉnh .
Ràng buộc
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 5 2 3 2 2 1 1 2 1 3 3 4 3 5 |
3 1 2 1 1 | Cây con đỉnh 1 gồm tất cả: màu {1,2,3} → 3. Đỉnh 3 gồm {2,2,1} → 2 màu. Đỉnh 2,4,5 chỉ có 1 màu. |
| 4 1 2 1 3 1 2 1 3 1 4 |
3 1 1 1 | Cây con đỉnh 1 chứa màu 1,2,3 → 3 màu phân biệt. |
Bình luận