原题链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal/
解题思路:
可以参考官方题解中“方法二:基于栈的遍历”的动画,结合我的题解和注释理解,动画的测试用例为[1,2,3,4,5,6]
。
- 对于整个二叉树而言,遍历的顺序是底部优先,其次是左侧优先。
- 对于一个最小单位的子树而言,遍历顺序依次是左子节点,根节点,右子节点。
/**
* @param {TreeNode} root
* @return {number[]}
*/
var inorderTraversal = function (root) {
let result = []; // 存储结果
let stack = []; // 使用栈进行遍历
let current = root; // 使用节点进行遍历
// 遍历的过程中,可能出现current为空,而stack中有元素,即为遍历到树的最底端。
// 也可能出现current有值而stack为空的情况,如遍历到二叉树的根节点。
while (current || stack.length) {
// 步骤1:不断向左下遍历节点,同时将所有节点入栈
while (current) {
stack.push(current);
current = current.left;
}
// 步骤2:当向左下方走到底之后,先取出最左下角的节点
current = stack.pop();
// 步骤3:保存左下角节点的值
result.push(current.val);
// 步骤4:遍历右节点。
// 如果此时右节点为空,则会回到步骤2,注意current会变为最近未取值的根节点。
// 以动画为例,如果步骤4的current为节点5,回到步骤2后,current会变为1,并且取值后就遍历到了节点3。
// 如果右节点不为空,则继续向下遍历。
current = current.right;
}
return result;
};