题目链接:
代码:
01. 前序(迭代)
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if (root == NULL) return result;
st.push(root);
while (!st.empty()) {
TreeNode* node = st.top(); // 中
st.pop();
result.push_back(node->val);
if (node->right) st.push(node->right); // 右(空节点不入栈)
if (node->left) st.push(node->left); // 左(空节点不入栈)
}
return result;
}
};
02. 后序(迭代)
class Solution {
public:
vector<int> inorderTraversal(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 {
cur = st.top(); // 从栈里弹出的数据,就是要处理的数据(放进result数组里的数据)
st.pop();
result.push_back(cur->val); // 中
cur = cur->right; // 右
}
}
return result;
}
};
03. 中序(迭代)
class Solution {
public:
vector<int> inorderTraversal(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 {
cur = st.top(); // 从栈里弹出的数据,就是要处理的数据(放进result数组里的数据)
st.pop();
result.push_back(cur->val); // 中
cur = cur->right; // 右
}
}
return result;
}
};
大概思路:
01. 前序(迭代)
中左右
注意要点:
代码里先右再左,这样输出才按照左右的顺序来(栈存储);
如果传入的根节点是null,则返回结果;
用栈来临时存储,最终存入数组里;
02. 后序(迭代)
左右中
用前序的代码,调换一下遍历左右的代码,反转下用函数就是后序。
03. 中序(迭代)
左中右
设置一个指针,向左遍历的同时记录值不为空的节点(临时储存在栈里),遇见null,使cur返回上一节点并弹出栈的元素,同时向右遍历。(循环条件:当指针为空以及栈为空的时候返回)
遇见的问题:
c++语法不熟悉{
01.result.push_back(node->val);
//往result传入node的值;
//push_back大概类似于栈的pop,队列的push;
}
个人想法:
这样的格式可以么,感觉有些地方风格不是很统一。刚开始接触树,有点难,但又不难。