标签:node 遍历 TreeNode nullptr st 二叉树 push 迭代法
中序遍历
中序遍历无法直接利用栈进行遍历,需要利用指针进行遍历,对栈中的节点进行操作。
对于中间节点,如果指针遍历到了,但没有进行处理,就再push()
一个nullptr
,作为标记,说明这个节点只是遍历过了,但是没有处理。事实上,每个待处理(放入vector)中的节点,其在栈中的上层元素都会是nullptr
#include <iostream>
#include <stack>
#include <vector>
using std::stack;
using std::vector;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
class Solution {
public:
vector<int> inorderTraversal(TreeNode *root) {
vector<int> result;
stack<TreeNode *> st;
if (root != nullptr)
st.push(root);
while (!st.empty()) {
TreeNode *node = st.top();
//对节点进行遍历
if (node != nullptr) {
st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
if (node->right != nullptr)
st.push(node->right);
st.push(node);
st.push(nullptr);
if (node->left != nullptr)
st.push(node->left);
} else {//依次处理节点
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
return result;
}
};
前序遍历
相比中序遍历,只是需要调整遍历部分,即st.push()
的顺序即可。
vector<int> preorderTraversal(TreeNode *root) {
vector<int> result;
stack<TreeNode *> st;
if (root != nullptr)
st.push(root);
while (!st.empty()) {
TreeNode *node = st.top();
if (node != nullptr) {
st.pop();
if (node->right != nullptr)
st.push(node->right);
if (node->left != nullptr)
;
st.push(node->left);
st.push(node);
st.push(nullptr);
} else {
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
}
后序遍历
相比中序遍历也只是调整遍历顺序而已
vector<int> postorderTraversal(TreeNode *root) {
vector<int> result;
stack<TreeNode *> st;
if (root != nullptr)
st.push(root);
while (!st.empty()) {
TreeNode *node = st.top();
if (node != nullptr) {
st.pop();
st.push(node);
st.push(nullptr);
if (node->right != nullptr)
st.push(node->right);
if (node->left != nullptr)
st.push(node->left);
} else {
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
}
标签:node,遍历,TreeNode,nullptr,st,二叉树,push,迭代法 来源: https://www.cnblogs.com/zwyyy456/p/16608515.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。