给定一个$n$个节点的树。现在你拥有$k$种颜色，你要用这些颜色给树上的每个节点染色，使得任何两个距离不大于2的不同节点所被染的颜色不同。

由于答案可能过大，请将其对$10^9+7$取模。