目录
之前参加一家公司面试被问到二叉树的相关非递归遍历算法,顿时懵逼,只记得递归算法的我瑟瑟发抖,下来之后就作了不少的功课。接下来要分享的,都是我经过对比之后选择的较为简单的非递归遍历方法,使用js语言。如有不足,还望指教。
假设一棵二叉树的结构为
一、前序遍历
使用栈的数据结构。初始化一个栈数组和结果数组
1.将根节点放入栈中, 栈:[1]
2.取出栈顶的节点,将其放入结果数组中, 结果数组: [1],栈: []
3.将取出的节点的右节点放入栈中, 栈: [3]
4.将取出的节点的左节点放入栈中,栈:[3,2]
5.回到步骤2(循环步骤2-4,直到栈为空)取出栈顶节点,将其放入结果数组中,结果数组:[1,2], 栈:[3]
6.将取出的节点的右节点放入栈中, 栈: [3,5]
7.将取出的节点的左节点放入栈中,栈: [3,5,4]
8.回到步骤2....
最后得到结果数组[1,2,4,5,3,6,7]
var preorderTraversal = function(root) {
let nodestack = [];
let res = [];
nodestack.push(root);
if(root===null){
return [];
}
while(nodestack.length>0){
let node = nodestack.pop();
res.push(node.val);
if(node.right){
nodestack.push(node.right);
}
if(node.left){
nodestack.push(node.left);
}
}
return res;
};
二、中序遍历
使用栈的数据结构。初始化一个栈数组和结果数组。
1.把根节点放入栈中,然后将根节点重新赋值为根节点的左节点,直到根节点为空。栈: [1,2,4],root = null
2.取出栈顶节点,并将其赋值为根节点,然后将其值放入结果数组