0

我有一个以空节点 a 开头的霍夫曼二叉树。

A 指向一个左节点和一个右节点，右节点也指向左节点和右节点。在拥有这棵树之后，是否可以递归地为每个节点设置父节点？

这是我在想的代码：

public Node setParents(Node n)
{
    if(n.getZero() == null && n.getOne() == null)
    {
        return n;
    }
    Node a = setParents(n.getZero()); // Zero being left
    a.setParent(n);
    Node b = setParents(n.getOne()); // One being right.
    b.setParent(n);
}

这是我当前如何通过使用优先级队列来创建树，该队列的值从最小到最大排序。

public Node createTree(PriorityQueue<Node> pq)
{

    while(pq.size() > 1)
    {
        Node n = new Node();

        Node a = pq.poll();
        if(pq.size() > 0)
        {
            Node b = pq.poll();
            n = new Node(a.getFrequency() + b.getFrequency());
            n.setZero(a);
            a.setWhich(0);
            a.setParent(n);
            n.setOne(b);
            b.setWhich(1);
            b.setParent(n);
        }
        else
        {
            n = new Node(a.getFrequency());
            n.setZero(a);
            a.setWhich(0);
            n.setParent(n);
            n.setOne(null);
        }
        pq.add(n);
    }
    Node n = pq.poll();
    n.setParent(null);
    setParents(n.getZero());
    setParents(n.getOne());
    return n;
}

我只需要确保每个节点都有一个父节点，我不知道从哪里开始。有什么帮助吗？

4

1 回答 1

0

对您的代码的一些注释可能会有所帮助

1. 不要在你的学习样本中使用 getter 和 setter 来进行简单的分配和读取，它们很难理解。

2. 如果您准备了一些物品，请不要将此物品与其他物品混合

        n.setZero(a);
        a.setWhich(0);
        a.setParent(n);
        n.setOne(b);

3. 据我了解有机会获得NPE

          if(pq.size() > 0) {
              Node b = pq.poll();
          }
     }
     Node n = pq.poll();
     n.setParent(null);      <-  n can be null

4. Java 有一个很好的特性，叫做 Enums

     a.setWhich(0);
     b.setWhich(1);

这是从根开始设置父母的方法

public void fixParents(Node parentNode)
{
    if (parentNode.zero != null) {
        parentNode.zero.parent = parentNode;
        fixParents(parentNode.zero);
    }
    if (parentNode.one != null) {
        parentNode.one.parent = parentNode;
        fixParents(parentNode.one);
    }
}

UPD

又一想。您在树构建功能中设置了父母。所以你应该只检查父母是否正确，而不是重新设置他们。

public void checkParents(Node parentNode) throws Exception
{
    if (parentNode.zero != null) {
        if (parentNode.zero.parent != parentNode) {
            throw new Exception( here include info about the parentNode.zero );
        }
        checkParents(parentNode.zero);
    }
    if (parentNode.one != null) {
        if (parentNode.one.parent != parentNode) {
            throw new Exception( here include info about the parentNode.one );
        }
        checkParents(parentNode.one);
    }
}
于 2013-04-17T06:48:04.107 回答