یک درخت n راسی به شما داده شده است که راسهای آن از 1 تا n شماره گذاری شده اند. هر راس در این درخت با یکی از رنگهای 1 تا k رنگآمیزی شده است.
مقدار مطلوبیت هر راس به شکل زیر تعریف میشود:
مقدار مطلوبیت تمام راسهای درخت را محاسبه کنید.
در خط اول ورودی دو عدد طبیعی n و k، نشاندهندهی تعداد راسهای درخت و حداکثر شمارهی رنگ راسهای درخت، آمده است.
در خط دوم ورودی n عدد طبیعی c1,c2,…,cn، نشاندهندهی رنگهای راسهای درخت، آمده است.
در n−1 خط بعدی ورودی، در هر خط دو عدد طبیعی vi و ui آمده است که نشاندهندهی وجود یک یال بین دو راس vi و ui است.
در تنها خط خروجی، 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 |