今天就跟大家聊聊有关如何根据前序遍历和中序遍历创建python二叉树,可能很多人都不太了解,为了让大家更加了解,小编给大家总结了以下内容,希望大家根据这篇文章可以有所收获。
前言四种遍历树的方法简介简介两种快速获得遍历结果的方法根据前序遍历和后续遍历创建树代码实现四种遍历树的方法的代码
不说废话了,下面讲讲如何根据pre_order和in_order创建二叉树。
首先这里简单介绍一下二叉树的4种遍历方式:
前序遍历(pre_order)
中序遍历(in_order)
后序遍历(post_order)
层序遍历(level_order)
至于这些遍历的代码放在文章的最后。
前序遍历就是先对当前的根节点进行操作,然后到左子节点,再到右子节点!
中序遍历就是先对当前左子节点进行操作,然后到当前根节点,再到右子节点!
后序遍历就是先对当前左子节点进行操作,然后到右子节点,再到当前根节点!
层序遍历就是按照从上到下从左到右的顺序对每个节点进行操作!代码写起来比前三个复杂点,得借助队列,并用迭代的方式来做。
如下图(之前上课做的笔记):
另外,介绍两个 可以快速地 根据树的形状 得出 前序、后序、中序 的遍历结果。
法一:
法二:
给你一个数组,用这个数组的值来创建一个树,结果有多种可能:
其中n是数组中元素的个数!
但是,如果我们给了两个数组,分别是前序遍历和后续遍历的结果,那么我们就能创建唯一的一个树!
Note:要求数组中的元素不重复,是唯一的!
看过上面对树的那几种遍历方式后,可以发现:
Note:下面的这个过程有点枯燥,我表述地也不太好,可以看后面的图。
前序遍历的第一个元素就是树的根节点;第二个元素是根节点的左子节点,这个左子节点也是后面的根节点;
根节点把中序遍历的数组一分为二,中序遍历的数组中:根节点的左边是左树,根节点的右边是右树
所以,我们就对前序遍历的数组进行遍历,当前索引记为pre_i,在每次遍历中,到中序遍历的数组中找这个pre_i对应的值,用这个pre_i把中序遍历的结果一分为二。这样往复下去就能还原树了。
下面我画一下整个流程:
大概就是这样,不断地对中序遍历的数组一分为二(根据前序遍历的数组的当前元素进行分割);中序遍历的数组的当前元素就是当前的根节点。
先定义树的节点:
template <class T>struct Node{ T val; Node* left; Node* right; explicit Node(T v) : val(v), left(nullptr), right(nullptr){}};
根据前序遍历和中序遍历创建树:
/** * @brief 根据前序遍历和中序遍历创建树 * @param[in] pre_vec 前序遍历的数组 * @param[in] in_vec 中序遍历的数组 * @param[in] left_in 中序遍历当前段的左边界 * @param[in] right_in 中序遍历当前段的右边界(超尾) * @static pre_i 前序遍历的当前的索引 * * @note 在每一层递归中,当前的 中序遍历 数组段 被分为:[left_in, pre_i), pre_i, [pre_i+1, right_in) * */template <class T>Node<T>* CreateTreeR(vector<T>& pre_vec, vector<T>& in_vec, int left_in, int right_in){ static int pre_i = 0; if (left_in < right_in) { /// 从 前序遍历 的数组中 获取 当前 根节点! T cur_root_val = pre_vec.at(pre_i); auto* cur_root = new Node<T>(cur_root_val); /// 遍历 中序遍历 的数组,找到当前根节点对应的索引 int i = left_in; while (i < right_in && cur_root_val != in_vec.at(i)) ++i; /// 下次递归前 pre_i 是需要向后移动一位的 ++pre_i; /// 一分为二!(注意,i是当前节点的索引哦!) cur_root->left = CreateTreeR(pre_vec, in_vec, left_in, i); /// 左树 cur_root->right = CreateTreeR(pre_vec, in_vec, i+1, right_in); /// 右树 return cur_root; } /// 当分到只剩最后一个元素时就返回空了 return nullptr;}
测试这段程序
对创建出来的这个树,用四种遍历方法分别遍历一下子,四种遍历的代码在文末。
上面这个是数字的,我现在拿字符串的试试:
1.层序遍历:
/** * @brief 树的层序遍历 * */ template<class T> void LevelOrder(Node<T>* root, vector<T>& vec){ /// 如果根节点为空就直接返回 if (!root) return; /// 定义一个队列放所有可能会成为 "根节点" 的节点(每次循环中都会pop出一个 "根节点" ) queue< Node<T>* > q; q.push(root); while (!q.empty()) { /// 从队列中拿出最前面的根节点 Node<T>* cur_root = q.front(); /// 这个变量在while外面声明更好,不用每次都创建一个新变量。 q.pop(); /// 保存当前 "根节点" 的值 vec.push_back(cur_root->val); /// 如果左子节点非空就把左子节点放入队列 if (cur_root->left) q.push(cur_root->left); /// 如果右子节点非空就把右子节点放入队列 if (cur_root->right) q.push(cur_root->right); }}
2.前序遍历:
/** * @brief 树的前序遍历 * */template<class T>void PreOrder(Node<T>* root, vector<T>& vec){ if (root) { vec.push_back(root->val); PreOrder(root->left, vec); PreOrder(root->right, vec); }}
3.中序遍历:
/** * @brief 树的中序遍历 * */template<class T>void InOrder(Node<T>* root, vector<T>& vec){ if (root) { InOrder(root->left, vec); vec.push_back(root->val); InOrder(root->right, vec); }}
4.后序遍历:
/** * @brief 树的后序遍历 * */template<class T>void PostOrder(Node<T>* root, vector<T>& vec){ if (root) { PostOrder(root->left, vec); PostOrder(root->right, vec); vec.push_back(root->val); }}
看完上述内容,你们对如何根据前序遍历和中序遍历创建python二叉树有进一步的了解吗?如果还想了解更多知识或者相关内容,请关注亿速云行业资讯频道,感谢大家的支持。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。
原文链接:https://my.oschina.net/MitchHong/blog/4504762