使用vector容器辅助二叉树先序遍历

使用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;
}

觉得我回答有用的话，记得点个关注哟！谢谢支持！