yukicoder No.1227 - I hate ThREE

No.1227 I hate ThREE - yukicoder 問題 $n$ 頂点の木がある。 $1$ 以上 $k$ 以下の整数からなる長さ $n$ の数列 $(p_i)$ で、以下を満たすものの個数を求めよ。 全ての辺 $uv$ について、 $|p_u - p_v| = 3$