二叉树的递归前中后序遍历
思路
1.首先要先确定递归函数的参数值和返回值
2.确定遍历的终止条件
3.确定单层遍历的逻辑
前序遍历
1.以力扣144.二叉树的前序遍历为例子,题目中给出了节点的定义以及函数的参数与返回值
void traversal(TreeNode* cur, vector<int>& vec)
2.要确定终止条件,在递归过程中,只要当前遍历的节点空了,单层递归就可以结束,直接返回上一次函数
``
if (cur == NULL) return;
``
3.接下来要确定单层的遍历逻辑,由于需要返回vector,所以单层遍历函数需要有两个参数,一个是节点,还有一个是返回的vector。只要节点不为空,就把节点值压入vector中,同时向左节点进发,直到左节点全部压入vector中,再转向右节点
void traversal(TreeNode* cur, vector<int>& vec) {
if (cur == NULL) return;
vec.push_back(cur->val); // 中
traversal(cur->left, vec); // 左
traversal(cur->right, vec); // 右
}
最后我们得到了最终代码:
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;
}
};
中序遍历和后序遍历
由此我们也可以写出中序遍历和后序遍历,只需要修改单层遍历函数中的逻辑
中序遍历:
void traversal(TreeNode* cur, vector<int>& vec) {
if (cur == NULL) return;
traversal(cur->left, vec); // 左
vec.push_back(cur->val); // 中
traversal(cur->right, vec); // 右
}
后序遍历:
void traversal(TreeNode* cur, vector<int>& vec) {
if (cur == NULL) return;
traversal(cur->left, vec); // 左
traversal(cur->right, vec); // 右
vec.push_back(cur->val); // 中
}
此时大家可以做一做leetcode上三道题目,分别是: