A14264 | Diameter Cuts

You are given an integer $k$ and an undirected tree, consisting of $n$ vertices.

The length of a simple path (a path in which each vertex appears at most once) between some pair of vertices is the number of edges in this path. A diameter of a tree is the maximum length of a simple path between all pairs of vertices of this tree.

You are about to remove a set of edges from the tree. The tree splits into multiple smaller trees when the edges are removed. The set of edges is valid if all the resulting trees have diameter less than or equal to $k$ .

Two sets of edges are different if there is an edge such that it appears in only one of the sets.

Count the number of valid sets of edges modulo $998\,244\,353$ .

The first line contains two integers $n$ and $k$ ( $2 \le n \le 5000$ , $0 \le k \le n - 1$ ) — the number of vertices of the tree and the maximum allowed diameter, respectively.

Each of the next $n-1$ lines contains a description of an edge: two integers $v$ and $u$ ( $1 \le v, u \le n$ , $v \neq u$ ).

The given edges form a tree.

Print a single integer — the number of valid sets of edges modulo $998\,244\,353$ .