为树中的叶子分配不同的值

Question

我正在尝试为树中的每一片叶子分配编号。

例如，如果我们有一棵有 6 片叶子的树，我希望叶子的数字从 0 到 5。

我不知道为什么我的代码不能很好地工作，我已经尝试了很多次，即使是递归方式，但似乎我错过了一些东西..

   public class Node {

    int index;
    int id;
    Node left;
    Node right;

    // Constructor and setters/getters.

    public static void num(Node n) {

    int ini=0;
    if(n==null)
    {

    }
    if(n.isLeaf())
    {
        n.index=ini;
        ini++;
    }
    if(!n.isLeaf())
    {
        num(n.getleft());
        num(n.getRight());
    }
}

我还想得到树上叶子的数量。

例如，我们的树看起来像

                                  1
                                /   \
                               2     3
                             /  \   / \ 
                            6    9  8  10
                           /
                          4
    public static int numberChild(Node n, int count)

{
    if (n == null) {
        return 0;
    }
    if (n.getleft() == null && n.getRight() == null) {
        return 1 + count;
    } else {
        int lc = numberChild(n.getleft(), count);
        int rc = numberChild(n.getRight(), lc);
        return rc;
    }

}

会给我错误的叶子数量，2 而不是 4！

有什么帮助吗？

Answer 1

如果 n == null 你返回 0。你应该返回 count 否则你之前的计数会丢失。在您的示例中，当计算 6 个节点的叶子时，您正确地从其下方的 4 个节点获得 1。然后你为不存在的右孩子调用numberChild()，它返回0，所以你为第6个节点返回的计数是0，而不是1。

编辑：要为叶子分配值我相信您应该使用与计算节点相同的想法，将计数传递给递归方法，以便它知道当前左侧已经编号了多少叶子节点，并返回更新的计数。您的 num 方法将类似于您的 numberChild 方法的另一个版本，通过将新索引分配给叶节点进行扩展。

Answer 2

刚刚发现我的分配索引代码出了什么问题

    public void num(Node n) {


        if(n.getleft()!=null)num(n.getleft());
        if(n.getRight()!=null)num(n.getRight());
        if(n.isLeaf())
        {
            n.assignIndex(ini);
            ini++;
        }

}

现在它运作良好:)。谢谢