Tổng màu thống trị trên cây
Đề bài
Mô tả
Cho một cây có gốc tại đỉnh , gồm đỉnh. Mỗi đỉnh được tô một màu.
Một màu được gọi là thống trị trong cây con của đỉnh nếu không có màu nào khác xuất hiện trong cây con của với số lần nhiều hơn . Vì vậy có thể có nhiều hơn một màu thống trị trong cùng một cây con.
Cây con của đỉnh gồm và tất cả các đỉnh sao cho đường đi từ tới gốc đi qua .
Với mỗi đỉnh , hãy tính tổng tất cả các màu thống trị trong cây con của .
Dữ liệu vào
- Dòng đầu chứa số nguyên — số đỉnh.
- Dòng thứ hai chứa số nguyên — màu của các đỉnh.
- Mỗi dòng trong dòng tiếp theo chứa hai số nguyên — một cạnh của cây.
Dữ liệu ra
In ra số nguyên — tổng các màu thống trị trong cây con của từng đỉnh, theo thứ tự từ đỉnh đến .
Ràng buộc
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 4 1 2 3 4 1 2 2 3 2 4 |
10 9 3 4 | Cây con của chứa cả đỉnh, mỗi màu xuất hiện đúng một lần nên tất cả đều thống trị, tổng . Cây con của là với màu , tổng . |
| 15 1 2 3 1 2 3 3 1 1 3 2 2 1 2 3 1 2 1 3 1 4 1 14 1 15 2 5 2 6 2 7 3 8 3 9 3 10 4 11 4 12 4 13 |
6 5 4 3 2 3 3 1 1 3 2 2 1 2 3 | Tại gốc , các màu đều xuất hiện lần nên đều thống trị, tổng . |
Bình luận