Kay và Bông Tuyết
Đề bài
Mô tả
Cho một cây có gốc gồm đỉnh, đánh số từ đến , gốc là đỉnh .
Cây con của đỉnh gồm đỉnh và tất cả các đỉnh nằm trong nhánh của (con cháu trực tiếp hoặc gián tiếp). Nói cách khác, cây con của gồm mọi đỉnh mà nằm trên đường đi từ về gốc.
Trọng tâm (centroid) của một cây (hoặc cây con) là một đỉnh mà nếu xóa nó đi thì mỗi thành phần liên thông còn lại có kích thước không vượt quá một nửa kích thước của cây (hoặc cây con) ban đầu.
Cho truy vấn. Với mỗi truy vấn gồm một đỉnh , hãy tìm một trọng tâm của cây con gốc . Nếu có nhiều đỉnh thỏa mãn, in ra đỉnh nào cũng được.
Dữ liệu vào
- Dòng đầu chứa hai số nguyên và : số đỉnh của cây và số truy vấn.
- Dòng thứ hai chứa số nguyên , trong đó là cha của đỉnh .
- dòng tiếp theo, mỗi dòng chứa một số nguyên : đỉnh xác định cây con cần tìm trọng tâm.
Dữ liệu ra
In ra dòng, dòng thứ là chỉ số của một trọng tâm của cây con gốc . Nếu có nhiều đáp án hợp lệ, in ra đáp án nào cũng được.
Ràng buộc
- , và dãy luôn xác định một cây hợp lệ.
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 7 4 1 1 3 3 5 3 1 2 3 5 |
3 2 3 5 |
Cây con gốc là toàn bộ cây (7 đỉnh); xóa đỉnh tạo ra các thành phần kích thước , đều không quá . Cây con gốc chỉ có một đỉnh nên trọng tâm là . Cây con gốc gồm hai đỉnh ; cả lẫn đều là trọng tâm hợp lệ, ở đây in . |
| 2 2 1 1 2 |
1 2 |
Cây con gốc có hai đỉnh ; xóa đỉnh để lại một thành phần kích thước nên là trọng tâm (đỉnh cũng hợp lệ). Cây con gốc chỉ có một đỉnh. |
Bình luận