使用vector容器辅助二叉树先序遍历
vector容器其实就是一个动态数组,不要被其名字吓到,其实很简单。
typedef int EmpeType
在本篇文章中使用EmpeType代表int类型。
typedef int EmpeType;
创建树的结构体
typedef struct TreeNode {
EmpeType val; //数据域
struct TreeNode* left; //左孩子
struct TreeNode* right; //右孩子
}TreeNode;
快速创建一个树
//快速创建一个树
TreeNode* A = (TreeNode*)malloc(sizeof(TreeNode));
A->val = 'A';
A->left = NULL;
A->right = NULL;
TreeNode* B = (TreeNode*)malloc(sizeof(TreeNode));
B->val = 'B';
B->left = NULL;
B->right = NULL;
TreeNode* C = (TreeNode*)malloc(sizeof(TreeNode));
C->val = 'C';
C->left = NULL;
C->right = NULL;
TreeNode* D = (TreeNode*)malloc(sizeof(TreeNode));
D->val = 'D';
D->left = NULL;
D->right = NULL;
TreeNode* E = (TreeNode*)malloc(sizeof(TreeNode));
E->val = 'E';
E->left = NULL;
E->right = NULL;
A->left = B;
A->right = C;
B->left = D;
B->right = E;
核心代码
void preOrder(TreeNode* root, vector<int>& res)
{
if (root == NULL)
return; //递归停止条件
res.push_back(root->val); //根结点进入容器
preOrder(root->left, res); //左孩子进入容器
preOrder(root->right, res); //右孩子进入容器
}
辅助配图理解代码
从根结点A开始入队。
A的左孩子不为空B入容器。
B的左孩子不为空,D入容器。
D的左右孩子为空,无法进入容器,这里的小递归结束,开始入B的右孩子。
B的右孩子不为空,E入容器。
E的左右孩子为空,无法进入容器,这里的小递归结束,开始入A的右孩子。
A的右孩子不为空,C入容器。
C的左右孩子为空,无法进入容器,这里的小递归结束,A结点遍历完毕,总的递归结束。
先序遍历的顺序都存到了vector容器中,遍历vector容器即可。
源代码
#include<iostream>
using namespace std;
#include<vector>
typedef int EmpeType;
typedef struct TreeNode {
EmpeType val; //数据域
struct TreeNode* left; //左孩子
struct TreeNode* right; //右孩子
}TreeNode;
void preOrder(TreeNode* root, vector<int>& res)
{
if (root == NULL)
return; //递归停止条件
res.push_back(root->val); //根结点进入容器
preOrder(root->left, res); //左孩子进入容器
preOrder(root->right, res); //右孩子进入容器
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
preOrder(root, res);
return res;
}
int main()
{
//快速创建一个树
TreeNode* A = (TreeNode*)malloc(sizeof(TreeNode));
A->val = 'A';
A->left = NULL;
A->right = NULL;
TreeNode* B = (TreeNode*)malloc(sizeof(TreeNode));
B->val = 'B';
B->left = NULL;
B->right = NULL;
TreeNode* C = (TreeNode*)malloc(sizeof(TreeNode));
C->val = 'C';
C->left = NULL;
C->right = NULL;
TreeNode* D = (TreeNode*)malloc(sizeof(TreeNode));
D->val = 'D';
D->left = NULL;
D->right = NULL;
TreeNode* E = (TreeNode*)malloc(sizeof(TreeNode));
E->val = 'E';
E->left = NULL;
E->right = NULL;
A->left = B;
A->right = C;
B->left = D;
B->right = E;
vector<int>ret = preorderTraversal(A);
cout << "Size = " << ret.size()<<endl;
//三种遍历vector容器的方式
for (int i = 0; i < ret.size(); ++i)
{
cout << (char)ret[i] << " ";
}
cout << endl;
for (auto it = ret.begin(); it != ret.end(); ++it)
{
cout << (char)*it << " ";
}
cout << endl;
for (vector<EmpeType>::iterator it = ret.begin(); it != ret.end(); ++it)
{
cout << (char)*it << " ";
}
return 0;
}
觉得我回答有用的话,记得点个关注哟!谢谢支持!