Leetcode[树] 94. 二叉树的中序遍历
程序员文章站
2024-01-11 17:08:34
...
审题
给定一个二叉树的根节点 root ,返回它的中序遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,3,2]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
示例 4:
输入:root = [1,2]
输出:[2,1]
示例 5:
输入:root = [1,null,2]
输出:[1,2]
中序遍历,对于递归算法来说很好想,也很好实现,但是本题要求的不是中序遍历输出元素,而是将元素保存在vector中,最后返回一个vector,因此,我们采用非递归遍历。
代码实现
方案一(上文提到的的方法):
这个和学二叉树的时候隔得时间太长了,到后来只能看回陈姥姥的PPT,小改了一下,抄了一段上来,不得不佩服人家这码力。
代码实现上,首先利用的是栈来模仿递归,因为要保存之前访问过的节点,整体的思路就是,遇到一个结点,就把它压栈,并去遍历它的左子树,当左子树遍历结束后,从栈顶弹出这个结点并访问它,然后按照其右指针去中序遍历该节点的右子树。(说人话)
好吧,其实就是采用了一个递归的思想, 首先遍历根节点的左子树(通过push(root)),怎么遍历根节点的左子树呢?通过遍历根节点的左子节点,怎么遍历根节点的左子节点呢?通过遍历根节点的左子节点的左子树->根->右子树…以此类推。
比如第一次循环,栈顶的元素应该是最左下的一个节点,因为没有左子树了,所以弹出他自己,然后遍历它的右子树…
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> v;
stack<TreeNode*> s;
while ( root || !s.empty() ) {
while ( root ) {
s.push(root);
root = root->left;
}
if ( !s.empty() ) {
root = s.top();
s.pop();
v.push_back(root->val);
root = root->right;
}
}
return v;
}
};
方案2:
哦哦,看完官方题解后发现,这道题也可以用递归,直接把vector当参数传进去就行了…是我太菜呜呜呜、
直接一气呵成,over。
class Solution {
public:
void inorderTraversal_para(TreeNode* root, vector<int>& v) {
if ( root == nullptr ) return;
inorderTraversal_para(root->left, v);
v.push_back(root->val);
inorderTraversal_para(root->right,v);
}
vector<int> inorderTraversal(TreeNode* root) {
vector<int> v;
inorderTraversal_para(root, v);
return v;
}
};
反思
树的遍历需要栈的相关知识和递归的用法,还挺综合的,不过对树的了解,我还是知之甚少,再看吧,再做两道题感受一下,今天先这样,收工。