C++ 中二叉树的最大宽度
问题陈述
给定一个二叉树,编写一个函数,以获取给定树的最大宽度。树的宽度是所有层级中最宽的一个。
考虑以下树−
10 / \ 7 4 / \ \ 9 2 1 / \ 2 5 1. Width at level 1: 1 2. Width at level 2: 2 3. Width at level 3: 3 4. Width at level 4: 2 For above tree answer is 3.
算法
1. Use level order traversal to find the answer
示例
#include <bits/stdc++.h> using namespace std; struct node { public: int data; node* left; node* right; }; int getWidth(node* root, int level); int height(node* node); node* newNode(int data); int getMaxWidth(node* root){ int maxWidth = 0; int width; int h = height(root); int i; for (i = 1; i <= h; ++i) { width = getWidth(root, i); if (width > maxWidth) { maxWidth = width; } } return maxWidth; } int getWidth(node* root, int level){ if (root == NULL) { return 0; } if (level == 1) { return 1; } else if (level > 1) { return getWidth(root->left, level - 1) + getWidth(root->right, level - 1); } } int height(node* node){ if (node == NULL) { return 0; } int lHeight = height(node->left); int rHeight = height(node->right); return (lHeight > rHeight)? (lHeight + 1): (rHeight + 1); } node* newNode(int data){ node* Node = new node(); Node->data = data; Node->left = NULL; Node->right = NULL; return(Node); } int main(){ node *root = newNode(10); root->left = newNode(7); root->right = newNode(4); root->left->left = newNode(9); root->left->right = newNode(2); root->right->right = newNode(1); root->right->right->left = newNode(2); root->right->right->right = newNode(5); cout<<"Maximum width = " << getMaxWidth(root) << endl; return 0; }
输出
编译并执行上述程序时。它将生成以下输出 −
Maximum width = 3
广告