Code Submission Evaluation System Login

Datatähti-valmennus

Väritys


Task | Statistics


CSES - Väritys

Time limit:1.00 s Memory limit:128 MB

Puussa on $n$ solmua, joista jokaisella on tietty väri väliltä $1 \ldots m$.

Tehtäväsi on laskea, mikä on suurin mahdollinen etäisyys kahden samanvärisen solmun välillä.

Syöte

Syötteessä on ensin kaksi kokonaislukua $n$ ja $m$: puun solmujen määrä ja värien määrä. Solmut on numeroitu kokonaisluvuin $1,2,\ldots,n$, ja värit on numeroitu kokonaisluvuin $1,2,\ldots,m$.

Sitten syötteessä on $n$ lukua $c_1,c_2,\ldots,c_n$, jotka kuvaavat kunkin solmun värin.

Lopuksi 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.

Voit olettaa, että puussa on kaksi solmua, jotka ovat samanvärisiä.

Tuloste

Tulosta suurin etäisyys kahden samanvärisen solmun välillä.

Rajat
Esimerkki

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


Tuloste:
2