DS查找—二叉树平衡因子

题目

问题 D: DS查找—二叉树平衡因子
时间限制: 1 Sec  内存限制: 128 MB
提交: 242  解决: 190
[提交][状态][讨论版]
题目描述
二叉树用数组存储，将二叉树的结点数据依次自上而下,自左至右存储到数组中，一般二叉树与完全二叉树对比，比完全二叉树缺少的结点在数组中用0来表示。

计算二叉树每个结点的平衡因子，并按后序遍历的顺序输出结点的平衡因子。

--程序要求--
若使用C++只能include一个头文件iostream；若使用C语言只能include一个头文件stdio
程序中若include多过一个头文件，不看代码，作0分处理
不允许使用第三方对象或函数实现本题的要求
输入
测试次数t

每组测试数据一行，数组元素个数n，后跟n个字符，二叉树的数组存储。

输出
对每组测试数据，按后序遍历的顺序输出树中结点的平衡因子（测试数据没有空树）

样例输入
2
6 ABC00D
24 ABCD0EF0000H00000000000I


样例输出
B 0
D 0
C 1
A -1
D 0
B 1
I 0
H 1
E 2
F 0
C 2
A -2

代码块

#include <iostream>
using namespace std;

class BiTree
{
private:
    int num;
    char *tree;
    void PostOrderTraverse(int p);
    void BalanceFactor(int p);
    int PreOrder(int p);
    void PreOrderTraverse(int p, int depth, int &maxdepth);
public:
    BiTree();
    ~BiTree();
    void PostOrder();
};

BiTree::BiTree()
{
    cin>>num;
    tree = new char[num+1];
    for(int i=1; i<=num; i++)
        cin>>tree[i];
}

BiTree::~BiTree()
{
    delete []tree;
}

void BiTree::PostOrder()
{
    PostOrderTraverse(1);
}

void BiTree::PostOrderTraverse(int p)
{
    if(tree[p]!='0' && p<=num)
    {
        PostOrderTraverse(2*p);
        PostOrderTraverse(2*p+1);
        BalanceFactor(p);
    }
}

void BiTree::BalanceFactor(int p)
{
    int temp;
    if(2*p>num)
        cout<<tree[p]<<" 0"<<endl;
    else if(2*p==num && 2*p+1>num)
        cout<<tree[p]<<" 1"<<endl;
    else
    {
        temp = PreOrder(2*p)-PreOrder(2*p+1);//平衡因子即为左子树深度减右子树深度
        cout<<tree[p]<<' '<<temp<<endl;
    }
}

int BiTree::PreOrder(int p)
{//外包装函数
    int depth = 0, maxdepth = 0;
    PreOrderTraverse(p, depth, maxdepth);
    return maxdepth;
}

void BiTree::PreOrderTraverse(int p, int depth, int &maxdepth)
{//遍历中求树的深度
    if(tree[p]!='0' && p<=num)
    {
        depth++;
        if(maxdepth<depth)
            maxdepth = depth;
        PreOrderTraverse(2*p, depth, maxdepth);
        PreOrderTraverse(2*p+1, depth, maxdepth);
    }
}

int main(void)
{
    int t;
    cin>>t;
    while(t--)
    {
        BiTree myTree;
        myTree.PostOrder();
    }
}