打印层次结构的二叉搜索树格式化

Question

我已经实现了以下代码，以按层顺序打印二叉搜索树。

public void printLevelOrder(int depth) {
    for (int i = 1; i <= depth; i++) {
        printLevel(root, i);
    }
}

public void printLevel(BinaryNode<AnyType> t, int level) {
    if (t == null) {
        return;
    }
    if (level == 1) {
        System.out.print(t.element);
    } else if (level > 1) {
        printLevel(t.left, level - 1);
        printLevel(t.right, level - 1);
    }
}

我正在尝试找出如何改进我的代码，使其以特定格式打印输出。

例如，给定一棵树：

    1 
   / \
  2   3
 /   / \
4   5   6

目前它的打印结果如下：

123456

我希望打印出以下内容：

Level 0: 1
Level 1: 2 3
Level 2: 4 5 6

- ILostMySpoon

---

让函数printLevel返回节点编号的字符串而不是立即打印它。然后，您可以以任何格式连接这些字符串。 - Aziz

Answer 1

不要在递归函数调用内立即打印值，使用字符串来保存值。这将使输出更易于操作。

public void printLevelOrder(int depth) {
    for (int i = 1; i <= depth; i++) {
        System.out.print("Level " + (i-1) + ": ");
        String levelNodes = printLevel(root, i);
        System.out.print(levelNodes + "\n");
    }
}

public String printLevel(BinaryNode<AnyType> t, int level) {
    if (t == null) {
        return "";
    }
    if (level == 1) {
        return t.element + " ";
    } else if (level > 1) {
        String leftStr = printLevel(t.left, level - 1);
        String rightStr = printLevel(t.right, level - 1);
        return leftStr + rightStr;
    }
    else // you need this to get it to compile
      return "";
}

输出：

Level 0: 1 
Level 1: 2 3 
Level 2: 4 5 6 

- Aziz