给定 $n, k$ 和一棵 $n$ 个节点的树，边权均为 $1$，问有多少个点对的距离为 $k$。

我只会 $O(nk)$，有什么更优的方法吗？