问题描述
给定一个二叉树,返回它的中序 遍历。
代码
思路:从根节点开始,一旦有左节点,就将父节点先入stack栈,直到叶子结点,将叶子结点入res栈,将该叶子结点的父节点出栈,并将父节点入res栈,找其右结点是否存在,不存在则回溯,继续出栈;存在,则将右结点当作新的根节点,重复开始的过程,直到stack栈为空,输出res栈。
var inorderTraversal = function(root) {
if(!root) return [];
let res = [], stack = [];
while(root) {
if(root.left){
stack.push(root);
root = root.left;
}else if(root.right){
res.push(root.val);
root = root.right;
}else if(!root.left&&!root.right){
res.push(root.val);
root = stack.pop();
root && (root.left = null);//注意这里,每次出栈后要判断root是否为空,空则循环结束
}
}
return res;
};