生成平衡二叉树

Generating a balanced binary tree

我不确定这是否在做它应该做的事情。 我的目标是从一组值生成一个平衡的二叉树。 请让我知道这是否正确。

注意:不是平衡二叉搜索树,只是平衡二叉树。

int heightPrivate(nodePtr node)
{
if (node == NULL)
  return -1;

return 1 + std::max(heightPrivate(node->left), heightPrivate(node->right));
} 

void addNodePrivate(nodePtr node, int val)
{
  if (root == NULL)
  {
    root = new BTNode;
    root->data = val;
    root->left = root->right = NULL;
  }
  else
  {
    if (node->left == NULL)
    {
      node->left = new BTNode;
      node->left->data = val;
      node->left->left = node->left->right = NULL;
    }
    else if (node->right == NULL)
    {
      node->right = new BTNode;
      node->right->data = val;
      node->right->left = node->right->right = NULL;
    }
    else
    {
      int lheight = heightPrivate(node->left);
      int rheight = heightPrivate(node->right);

      if (lheight < rheight)
        addNodePrivate(node->left, val);
      else if (rheight < lheight)
        addNodePrivate(node->right, val);
      else
        addNodePrivate(node->left, val);
    }
  }
}

void printPostorderPrivate(nodePtr p, int indent=0)
{
  if(p != NULL) {
    if(p->left) printPostorderPrivate(p->left, indent+4);
    if(p->right) printPostorderPrivate(p->right, indent+4);
    if (indent) {
        std::cout << std::setw(indent) << ' ';
    }
    std::cout<< p->data << " \n ";
  }
}

主要...

int main()
{
  BTree tree;

  tree.addNode(1);
  tree.addNode(2);
  tree.addNode(3);
  tree.addNode(4);
  tree.addNode(5);
  tree.addNode(6);
  tree.addNode(7);

  tree.printPostorder();

我得到的结果是这样的:

            7
         4 
         6
     2
         5
     3
 1

2的children是4和5,问题是为什么7会更上一层楼

7 出现在它所在位置的原因是因为在 addNodePrivate 方法中检查两个 child 分支的高度,如果它们相等则向左。

因此,当您插入 7 时,当程序位于根节点(节点 1)时,它看到左分支的高度和右分支的高度都等于 1(节点 2 有 children 5 和 4 但没有 grandchildren,节点 3 有 child6 也没有 grandchildren),所以它向左 - 沿着节点 2 的分支。

要达到你想要的效果,你需要选择最短路径的分支,所以比较两个分支的高度是不够的。

希望对您有所帮助,祝您好运。