二叉树的高度和宽度(C++)

最新推荐文章于 2024-08-14 19:29:15 发布

赵延超

最新推荐文章于 2024-08-14 19:29:15 发布

阅读量1k

点赞数

分类专栏： ++++++++Algorithm++++++ # C++

本文链接：https://blog.csdn.net/SoftpaseFar/article/details/109274035

版权

++++++++Algorithm++++++ 同时被 2 个专栏收录

82 篇文章 0 订阅

订阅专栏

C++

81 篇文章 4 订阅

订阅专栏

详细代码

#include <iostream>
#include <queue>
#include <cmath>


using namespace std;

struct TreeNode {
    int val;
    TreeNode *left, *right;

    TreeNode(int x = 0, TreeNode *l = NULL, TreeNode *r = NULL) : val(x), left(l), right(r) {}
};

//获取树的高度
int GetDepth(TreeNode *t) {
    if (!t) {
        return 0;
    }
    int ld = GetDepth(t->left);
    int rd = GetDepth(t->right);
    return max(ld, rd) + 1;
}

//获取树的宽度
int GetWidth(TreeNode *t) {
    if (!t) {
        return 0;
    }
    queue<TreeNode *> Q;
    int maxWidth = 1;
    Q.push(t);

    while (1) {
        int len = Q.size();
        if (!len) {
            break;
        }
        while (len--) {

            TreeNode *temp = Q.front();
            Q.pop();
            if (temp->left) {
                Q.push(temp->left);
            }
            if (temp->right) {
                Q.push(temp->right);
            }
        }
        maxWidth = max(maxWidth, (int) Q.size());
    }
    return maxWidth;
}

int main() {

    TreeNode t8(8);
    TreeNode t9(9);
    TreeNode t4(4);
    TreeNode t5(5, &t8, NULL);
    TreeNode t6(6, NULL, &t9);
    TreeNode t7(7);
    TreeNode t2(2, &t4, &t5);
    TreeNode t3(3, &t6, &t7);
    TreeNode t1(1, &t2, &t3);
    TreeNode *T = &t1;

    //输出树的高度
    cout << GetDepth(T) << endl;
    //输出树的宽度
    cout << GetWidth(T) << endl;

    return 0;
}