یک درخت $n$ راسی به شما داده شده است که راسهای آن از $1$ تا $n$ شماره گذاری شده اند. هر راس در این درخت با یکی از رنگهای $1$ تا $k$ رنگآمیزی شده است.
مقدار مطلوبیت هر راس به شکل زیر تعریف میشود:
مقدار مطلوبیت تمام راسهای درخت را محاسبه کنید.
در خط اول ورودی دو عدد طبیعی $n$ و $k$، نشاندهندهی تعداد راسهای درخت و حداکثر شمارهی رنگ راسهای درخت، آمده است.
در خط دوم ورودی $n$ عدد طبیعی $c_1, c_2, \dots, c_n$، نشاندهندهی رنگهای راسهای درخت، آمده است.
در $n-1$ خط بعدی ورودی، در هر خط دو عدد طبیعی $v_i$ و $u_i$ آمده است که نشاندهندهی وجود یک یال بین دو راس $v_i$ و $u_i$ است.
در تنها خط خروجی، $n$ عدد چاپ کنید که عدد $i$ام مقدار مطلوبیت راس $i$ام را نشان میدهد.
ورودی نمونه | خروجی نمونه |
---|---|
5 2 1 1 2 1 1 1 2 2 3 3 4 4 5 | 2 3 2 3 2 |
7 4 1 2 4 1 1 1 1 1 2 1 3 2 4 2 5 3 6 3 7 | 4 4 4 3 3 3 3 |