leetcode 94.二叉树的中序遍历
题目描述
给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
解题思路
递归: 中序遍历的顺序是左->根->右,直接使用递归函数,代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
if (root == NULL){
return res;
}
inorderTraversal(root->left);
res.push_back(root->val);
inorderTraversal(root->right);
return res;
}
vector<int> res;
};
迭代: 使用栈按照中序遍历的思路去存储节点,栈的性质是先进后出,那么在遍历左节点的过程中,不断的压栈,直到左节点为空,弹出栈顶元素,并将节点的值添加到vector中,然后去查找当前节点的右节点。下面一个动图可以更加容易理解该种方法的原理。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> treeNode; // 栈,存储树中的节点
TreeNode* cur = root; // 移动指针
vector<int> res; // 存储每个节点的value
while(cur != NULL || !treeNode.empty()){
// 不断的在当前节点下,查找左子树
while(cur){
treeNode.push(cur);
cur = cur->left;
}
cur = treeNode.top(); // 左子树查找完毕
treeNode.pop(); // 弹出节点
res.push_back(cur->val); // cur是叶节点就是访问左节点,若是cur为不为空,则是访问根节点
cur = cur->right; // 进入右节点
}
return res;
}
};
欢迎大家关注我的个人公众号,同样的也是和该博客账号一样,专注分享技术问题,我们一起学习进步