解决三个题目:前中后序的遍历
二叉树的前序遍历
题目:
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
二叉树的后序遍历
题目:
给你一棵二叉树的根节点 root ,返回其节点值的 后序遍历 。
二叉树的中序遍历
题目:
给定一个二叉树的根节点 root ,返回它的 中序 遍历。
代码随想录
难以整合的原因:
中序无法同时解决访问节点(遍历节点)和处理节点(将元素放进结果集)不一致的情况。
思路分为两部分:
存储元素:右中左,出栈然后右中左然后一直这样,此时就是把左边的全部处理完(即先处理左树)。
回收元素:左边到顶了下一个为null了就回收元素,然后节点到中节点,此时也不进行右中左,节点到右节点时,此时继续进行右中左,右中左。(即将右节点当做根节点处理右树)。
解决访问节点(遍历节点)和处理节点(将元素放进结果集)不一致的情况:
从栈中取元素作为根节点,然后根据不同类型的跟节点进行不同的操作。
对左节点的操作以及对右节点的操作都是继续搜索(即继续右中左)对根节点的操作当然不需要右中左了(那么怎么判断到达根节点了呢,通过栈中根节点下一个元素为null来判断)
中序遍历代码如下:
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
if (root != NULL) st.push(root);
while (!st.empty()) {
TreeNode* node = st.top();
if (node != NULL) {
st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
if (node->right) st.push(node->right); // 添加右节点(空节点不入栈)
st.push(node); // 添加中节点
st.push(NULL); // 中节点访问过,但是还没有处理,加入空节点做为标记。
if (node->left) st.push(node->left); // 添加左节点(空节点不入栈)
} else { // 只有遇到空节点的时候,才将下一个节点放进结果集
st.pop(); // 将空节点弹出
node = st.top(); // 重新取出栈中元素
st.pop();
result.push_back(node->val); // 加入到结果集
}
}
return result;
}
};
前序遍历代码如下:
这就是按照右左中的顺序,第一个出栈即出中节点不进行右左中,当为左节点、右节点时继续进行右左中。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
if (root != NULL) st.push(root);
while (!st.empty()) {
TreeNode* node = st.top();
if (node != NULL) {
st.pop();
if (node->right) st.push(node->right); // 右
if (node->left) st.push(node->left); // 左
st.push(node); // 中
st.push(NULL);
} else {
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
return result;
}
};
后续遍历代码如下:
这就是按照中右左的顺序,第一个第二个出栈分别为出左节点右节点都进行中右左的操作,当为中节点时不进行中右左。
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
if (root != NULL) st.push(root);
while (!st.empty()) {
TreeNode* node = st.top();
if (node != NULL) {
st.pop();
st.push(node); // 中
st.push(NULL);
if (node->right) st.push(node->right); // 右
if (node->left) st.push(node->left); // 左
} else {
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
return result;
}
};