問題:
給定二叉樹的初始化數(shù)據(jù)腺怯,怎樣動態(tài)建立一個二叉樹呢川无?
比如我們給定這樣的一組數(shù)據(jù):{ 1, 2, 3, 4, 0, 5, 6, 0, 7 }(假設(shè)0代表空)懦趋,則我們構(gòu)建的二叉樹是這樣的:
1
/ \
2 3
/ /
4 5 6
7
思路分析:
我們可以使用一個隊列,隊首出一個元素仅叫,隊未進兩個元素,而這兩個元素正好是這個隊首元素的左右節(jié)點笙隙。
頭文件 (TreeNode.h)
#pragma once
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
//初始化一個二叉樹
TreeNode* initBTree(int elements[], int size);
//樹的前序遍歷
void preOrder(TreeNode *root, vector<int> &result);
//樹的中序遍歷
void inOrder(TreeNode *root, vector<int> &result);
//樹的后序遍歷
void postOrder(TreeNode *root, vector<int> &result);
/*樹的深度*/
int btreedepth(TreeNode *bt);
/*樹的節(jié)點數(shù)*/
int btreecount(TreeNode *bt)
//vector的遍歷
void traverse(vector<int> nums);
函數(shù)(TreeNode.cpp)
#include "TreeNode.h"
/*初始化一個二叉樹*/
TreeNode* initBTree(int elements[], int size)
{
if (size < 1)
{
return NULL;
}
//動態(tài)申請size大小的指針數(shù)組
TreeNode **nodes = new TreeNode*[size];
//將int數(shù)據(jù)轉(zhuǎn)換為TreeNode節(jié)點
for (int i = 0; i < size; i++)
{
if (elements[i] == 0)
{
nodes[i] = NULL;
}
else
{
nodes[i] = new TreeNode(elements[i]);
}
}
queue<TreeNode*> nodeQueue;
nodeQueue.push(nodes[0]);
TreeNode *node;
int index = 1;
while (index < size)
{
node = nodeQueue.front();
nodeQueue.pop();
nodeQueue.push(nodes[index++]);
node->left = nodeQueue.back();
nodeQueue.push(nodes[index++]);
node->right = nodeQueue.back();
}
return nodes[0];
}
/*樹的前序遍歷*/
void preOrder(TreeNode *root, vector<int> &result)
{
if (root)
{
result.push_back(root->val);
preOrder(root->left, result);
preOrder(root->right, result);
}
}
/*樹的中序遍歷*/
void inOrder(TreeNode *root, vector<int> &result)
{
if (root)
{
inOrder(root->left, result);
result.push_back(root->val);
inOrder(root->right, result);
}
}
/*樹的后序遍歷*/
void postOrder(TreeNode *root, vector<int> &result)
{
if (root)
{
postOrder(root->left, result);
postOrder(root->right, result);
result.push_back(root->val);
}
}
/*樹的深度*/
int btreedepth(TreeNode *bt)
{
if (bt == NULL)
return 0;
else
{
int dep1 = btreedepth(bt->left);
int dep2 = btreedepth(bt->right);
if (dep1>dep2)
return dep1 + 1;
else
return dep2 + 1;
}
}
/*樹的節(jié)點數(shù)*/
int btreecount(TreeNode *bt)
{
if (bt == NULL)
return 0;
else
return btreecount(bt->left) + btreecount(bt->right) + 1;
}
/*樹的葉子樹*/
int btreeleafcount(TreeNode *bt)
{
if (bt == NULL)
return 0;
else if (bt->left == NULL&&bt->right == NULL)
return 1;
else return btreeleafcount(bt->left) + btreeleafcount(bt->right);
}
/*輸出樹*/
void printbtree(TreeNode *bt)
{
if (bt == NULL)
return;
else
{
cout << bt->val;
if (bt->left != NULL || bt->right != NULL)
{
cout << '(';
printbtree(bt->left);
if (bt->right != NULL)
cout << ',';
printbtree(bt->right);
cout << ')';
}
}
}
/*Vector遍歷*/
void traverse(vector<int> nums)
{
vector<int>::size_type size = nums.size();
for (size_t i = 0; i < size; i++)
{
cout << nums[i] << ' ';
}
cout << endl;
}