题目描述
给定一个二叉树的根节点root,返回它的中序遍历 。
问题分析
递归法: 中序遍历是左
→
\rightarrow
→中
→
\rightarrow
→右的顺序,所以递归逻辑是先处理左子树,再取中节点的数值,最后处理右子树。
迭代法: 中序遍历的迭代法就没前序遍历那么简单了。因为先访问的是二叉树顶部的节点,然后一层一层向下访问,直到达左面的底部,再开始处理节点(也就是把节点的数值放入result数组),这就造成访问顺序和处理顺序是不一致的。解决方法是需要借助指针的来访问节点,使用栈处理节点上的元素。
二叉树的前序遍历
代码实现
// 编程软件:VS2019
// 参考书籍:代码随想录
#include<iostream>
#include<vector>
#include<stack>
using namespace std;
// 定义二叉树
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int v):val(v),left(nullptr),right(nullptr){}
};
// 递归法
void traversal(TreeNode* cur, vector<int>& vec) {
if (cur == NULL) return;
traversal(cur->left, vec); // 递归左孩子
vec.push_back(cur->val); // 当前节点
traversal(cur->right, vec); // 递归右孩子
}
vector<int> inorderTraversal1(TreeNode* root) {
vector<int> result;
traversal(root, result);
return result;
}
// 迭代法
vector<int> inorderTraversal2(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
TreeNode* cur = root;
while (cur != NULL || !st.empty()) {
if (cur != NULL) { // 指针访问节点,访问到底层
st.push(cur); // 将访问的节点放入栈
cur = cur->left; // 左孩子
}
else {
// 从栈里弹出的数据就是要处理的数据(放入result数组的数据)
cur = st.top();
st.pop();
result.push_back(cur->val); // 父节点
cur = cur->right; // 右孩子
}
}
return result;
}
// 输出数组v中元素的值
void print(vector<int> v) {
for (int i = 0; i < v.size(); i++) {
cout << v[i] << " ";
}
cout << endl;
}
int main() {
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->left = new TreeNode(4);
root->left->right = new TreeNode(5);
root->left->left->left = new TreeNode(6);
vector<int> res;
res = inorderTraversal1(root);
print(res);
}
// 结果:6 4 2 5 1 3