前言
二叉树的题目开刷!以下是题目分类,今天写遍历这部分
递归三要素
-
确定递归函数的参数和返回值: 确定哪些参数是递归的过程中需要处理的,那么就在递归函数里加上这个参数, 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。
-
确定终止条件: 写完了递归算法, 运行的时候,经常会遇到栈溢出的错误,就是没写终止条件或者终止条件写的不对,操作系统也是用一个栈的结构来保存每一层递归的信息,如果递归没有终止,操作系统的内存栈必然就会溢出。
-
确定单层递归的逻辑: 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。
二叉树的递归遍历
-
// 前序遍历 class Solution { public: void traversal(TreeNode* cur, vector<int>& vec) { if (cur == NULL) return; // 终止条件 vec.push_back(cur->val); // 中 traversal(cur->left, vec); // 左 traversal(cur->right, vec); // 右 } vector<int> preorderTraversal(TreeNode* root) { vector<int> result; traversal(root, result); return result; } }; // 中序和后序类似,调换左中右顺序
二叉树的迭代遍历
- 用栈实现,注意先进右孩子再进左孩子
-
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; } };
- 和前序的代码差不多,左右孩子入栈顺序调换后,再reverse一下数组就是后序了,妙
-
class Solution { public: vector<int> postorderTraversal(TreeNode* root) { stack<TreeNode*> st; vector<int> res; if(root == nullptr) return res; st.push(root); while(!st.empty()){ TreeNode* tmp = st.top(); st.pop(); res.push_back(tmp->val); if(tmp->left) st.push(tmp->left); if(tmp->right) st.push(tmp->right); } reverse(res.begin(), res.end()); return res; } };
- 中序遍历的遍历顺序和处理顺序不同
-
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; } };
二叉树的统一迭代法
- 中序遍历案例,用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; } };
后言
层序遍历的题好多,明天晚上再刷。特别想让事情可控并且固定在每天的某个时候,但是总有很多意外导致每天少了几样,结果就好几天都没刷题了,还是要规律起来呀(只要没啥要紧事)