Code Submission Evaluation System Login

Algoritmit ongelmanratkaisussa 2019

Eri arvot


Task | Statistics


CSES - Eri arvotCSES - Eri arvot

Time limit:1.00 s Memory limit:512 MB

Sinulle annetaan juurellinen puu, jonka jokaisessa solmussa on tietty arvo. Tehtäväsi on laskea jokaiseen solmuun, montako eri arvoa solmun alipuussa on.

Syöte

Syötteen ensimmäisellä rivillä on kokonaisluku $n$: solmujen määrä. Solmut on numeroitu $1,2,\dots,n$, ja solmu $1$ on puun juuri.

Seuraavalla rivillä on $n$ kokonaislukua $x_1,x_2,\dots,x_n$: kunkin solmun arvo.

Sitten syötteessä on $n-1$ riviä, jotka kuvaavat puun rakenteen. Jokaisella rivillä on kaksi kokonaislukua $a$ ja $b$: solmujen $a$ ja $b$ välillä on kaari.

Tuloste

Tulosta $n$ kokonaislukua: vastaus jokaiselle solmulle.

Rajat
Esimerkki

Syöte:
5
2 1 3 1 2
1 2
1 3
3 4
3 5


Tuloste:
3 1 3 1 1