c# truncate the tree, keeping the specified number of nodes

TruncateByTraverseBF.cs

public Node TruncateByTraverseBF(Node root,int count)
{
    if (count <= 0)
        return null;

    if (count == 1)
    {
        root.Children.Clear();
        return root;
    }

    var queue = new Queue<Node>();
    queue.Enqueue(root);

    var currentLength = 1;

    int i = 0;
    while (queue.TryDequeue(out Node subTree))
    {
        for (i = 0; i < subTree.Children.Count && currentLength < count; i++, currentLength++)
            queue.Enqueue(
                subTree.Children[i]);

        if (currentLength >= count)
        {
            subTree.Children.RemoveRange(i, subTree.Children.Count - i);

            break;
        }
    }

    while (queue.TryDequeue(out Node subTree))
        subTree.Children.Clear();

    return root;
}



/*
 * Example
 */
 
                         /31-41           /31
                      /21-32           /21
TruncateByTraverseBF(1-22-33   , 5) = 1-22
                      \23              \23