Bộ sưu tập màu
Xem dạng PDF📖 Bài giảng & lời giải: đây là Bài tập 4 trong bài giảng Bài 44: Small-to-large, Euler tour trên blog Lập trình thuật toán – có lý thuyết, mô phỏng từng bước, gợi ý và code C++ / Python.
⚠️ Lưu ý: hãy đảm bảo em đã thật sự cố gắng tự làm bài trước, rồi mới xem gợi ý và lời giải nhé!
🎓 Khóa học Lập trình Thuật toán thi đấu của thầy Nguyễn Việt Nam Sơn – 111 buổi video, kèm cặp cá nhân, luyện thi chuyên Tin · HSG Tin học · Tin học trẻ · đội tuyển quốc gia · Olympic/ICPC. 👉 Xem chi tiết khóa học · Thành tích học viên
📞 Đăng ký học: Zalo 0767 666 702 · Facebook Nguyễn Việt Nam Sơn · 🌐 laptrinhthuattoan.blogspot.com
Cho cây $n$ đỉnh, đánh số $1..n$, gốc là đỉnh $1$; đỉnh $u$ được tô màu $c_u$. Với mỗi đỉnh $u$, hãy đếm số màu khác nhau xuất hiện trong cây con gốc $u$ (kể cả $u$).
Dữ liệu vào
- Dòng 1: số nguyên $n$ ($1 \le n \le 2 \cdot 10^5$).
- Dòng 2: $n$ số nguyên $c_1, c_2, \ldots, c_n$ ($1 \le c_u \le 10^9$).
- $n - 1$ dòng tiếp theo, mỗi dòng gồm hai số nguyên $u, v$ ($1 \le u, v \le n$, $u \ne v$) là một cạnh của cây (dữ liệu đảm bảo $n - 1$ cạnh tạo thành một cây).
Kết quả
In ra trên một dòng $n$ số nguyên, số thứ $u$ là số màu khác nhau trong cây con gốc $u$, các số cách nhau một dấu cách.
Sample Input
10
1 2 1 3 2 3 1 2 1 3
1 2
1 3
2 4
2 5
3 6
3 7
5 8
5 9
7 10
Sample Output
3 3 2 1 2 1 2 1 1 1
Bình luận