给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
1
2
/
3
输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
递归法
c++
/**
* 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 {
vector<int>res;//注意初始化res的位置;
//当放进inorderTraversal()函数中会递归一次就会输出,这样的结果是错误的
public:
vector<int> inorderTraversal(TreeNode* root) {
if(root){
inorderTraversal(root->left);
res.push_back(root->val);
inorderTraversal(root->right);
}
return res;
}
};
迭代法(运用栈)
/**
* 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) {
vector<int>res;
stack<TreeNode>s;
TreeNode* t=root;
while(t||!s.empty()){
//一直遍历到叶子节点;
while(t){
s.push(*t);//将所有的左节点进栈,这些节点包括原来认为是根节点
t = t->left;
}
if(!s.empty()){
res.push_back(s.top().val)//;将左节点和根节点的值进res;
t=s.top().right;//找右节点;
s.pop();//导出这个节点
}
}
return res;
}
};
再补充一个更新的解法,python3:
其核心思想如下:
使用颜色标记节点的状态,新节点为白色,已访问的节点为灰色。
如果遇到的节点为白色,则将其标记为灰色,然后将其右子节点、自身、左子节点依次入栈。
如果遇到的节点为灰色,则将节点的值输出。
使用这种方法实现的中序遍历如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def inorderTraversal(self, root: TreeNode) -> List[int]:
black,white = 1,0
res = []
stack = [(white,root)]
while stack:
color,node = stack.pop()
if node is None:
continue
if color==white:
stack.append((white,node.right))
stack.append((black,node))
stack.append((white,node.left))
else:
res.append(node.val)
return res
如要实现前序、后序遍历,只需要调整左右子节点的入栈顺序即可。