1448 - 色付き山小屋
時間制限 3 秒 / メモリ制限 256 MB / 得点 13 / Writer syoribu / x 3 / 統計 /
-
タグ:
- PCK予選_12問目
- PCK2020予選
問題文
あなたは$N$棟の山小屋と、山小屋の間をつなぐ道を管理しています。山小屋にはそれぞれ$1$から$N$までの番号がついています。道は全部で$N−1$本あります。どの道も同じ長さで、どちらの方向にも通ることができます。また、どの山小屋からでも、いくつかの道を通っていくことで他のどの山小屋にも到達できます。すべての山小屋には目印となる色が塗ってあります。登山者が同じ色の山小屋の間を移動できるように、どの山小屋にもその色と同じ色をした他の山小屋が少なくとも1つあります。
あなたは、それぞれの山小屋について、その山小屋と同じ色の他の山小屋の中で最短の移動距離で到達できる山小屋を求めようとしています。
課題
山小屋の色と、山小屋同士を直接つなぐ道の情報が与えられる。それぞれの山小屋について、その山小屋と同じ色の他の山小屋の中から移動距離が最短のものを見つけ、その最短の移動距離を出力するプログラムを作成せよ。ただし、道の長さはすべて1とする。
入力
入力は以下の形式で与えられる。
$N$ $c_{1}$ $c_{2}$ $\vdots$ $c_{N}$ $s_{1} \ t_{1}$ $s_{2} \ t_{2}$ $\vdots$ $s_{N-1} \ t_{N-1}$
1行目に山小屋の数$N\ (1 \leq N \leq 200,000)$が与えられる。続く$N$行に$i$番目の山小屋の色$c_{i}$が整数で与えられる。
続く$ \ N \ $行に$ \ i \ $番目の山小屋の色$ \ c_i \ (1 \leq c_i \leq N) \ $が整数で与えられる。続く$N-1$行に、2つの山小屋を直接つなぐ道の情報が与えられる。$s_{i}$と$t_{i}(1 \leq s_{i} \lt t_{i} \leq N)$は$i$番目の道がつなぐ2つの山小屋の番号である。ただし、どの2つの山小屋についても、それらを直接つなぐ道は1本以下とする。
出力
出力は$N$行である。山小屋$i$について、同じ色の他の山小屋の中から移動距離が最短のものを見つけ、その最短の移動距離を$i$行目に出力する。
入出力例
入力例1
8 1 2 3 3 2 2 3 1 1 2 1 3 2 4 2 5 4 6 5 7 7 8
出力例1
4 1 3 3 1 2 3 4