DSU Trên Cây (Sack) - Đếm Số Màu Phân Biệt Trên Cây Con
Cho một cây có gốc gồm N đỉnh (đánh số từ 1 đến N, gốc là đỉnh 1). Mỗi đỉnh u được gán một màu nguyên dương Cu.
Tiến độ của tôi ở bài này
Điểm được lưu vào tài khoản sau khi chấm bài.
Đang tải điểm của bạn…
Kiến thức và chủ đề
Kiến thức tiên quyết: Heavy-Light Decomposition Concept, Tree Subtree Size, Map / Frequency Array.
Nội dung đề bài
Mô tả bài toán
Cho một cây có gốc gồm N đỉnh (đánh số từ 1 đến N, gốc là đỉnh 1). Mỗi đỉnh u được gán một màu nguyên dương Cu.
Với mỗi đỉnh u (1 ≤ u ≤ N), hãy đếm số lượng màu phân biệt xuất hiện trong cây con gốc u (bao gồm chính đỉnh u và tất cả các con cháu của nó).
Quy ước nộp bài
- Chỉ cần viết một chương trình đọc
stdinvà in rastdout. Bài này không yêu cầu viết hàm. - Không dùng
coutđể in lời nhắc trước khi đọc dữ liệu. Lời nhắc sẽ lọt vàostdoutvà làm bài sai. - Chỉ in đúng nội dung ở mục Output. Không in thêm nhãn, dòng trống hay ký tự thừa.
- Output được so khớp từng ký tự, phân biệt chữ hoa/thường và dấu câu.
Input
- Dòng đầu chứa số nguyên N (1 ≤ N ≤ 2 · 105).
- Dòng thứ hai chứa N số nguyên C1, C2, …, CN (1 ≤ Ci ≤ 109).
- N - 1 dòng tiếp theo, mỗi dòng chứa hai số nguyên u, v (1 ≤ u, v ≤ N) biểu diễn một cạnh của cây.
Output
- In ra N số nguyên trên một dòng, cách nhau bởi dấu cách: số lượng màu phân biệt trong cây con của các đỉnh 1, 2, …, N.
Ràng buộc
- Thời gian chạy tối đa: 1000ms.
- Giới hạn bộ nhớ: 256MB.
- Dữ liệu đầu vào tuân thủ đúng định dạng và miền giá trị được mô tả.
Ví dụ 1
Input
5
2 3 2 2 1
1 2
1 3
3 4
3 5Output
3 1 2 1 1Giải thích
Dữ liệu kiểm thử mẫu và kết quả thực thi theo yêu cầu.
Gợi ý và lời giải chỉ mở sau khi bạn bấm Nộp bài. Giáo viên và quản trị viên mở được ngay.
Góp ý & báo lỗi bài tập
Đề bài chưa rõ, test có vấn đề hay bạn có ý tưởng giúp bài tốt hơn? Gửi cho đội ngũ AI Empire nhé — mỗi góp ý đều được đọc.