javascript 遍历有序BST

Posted 2021-05-03

/* input: [2,3,4,5,2,3,4]
          2
        /   \
       3     4
      / \   / \
     5   2 3   4
*/
const input = {
  val: 2,
  right: {
     val: 4,
     right: { val: 4, right: null, left: null },
     left: { val: 3, right: null, left: null }
  },
  left: {
     val: 3,
     right: { val: 2, right: null, left: null },
     left: { val: 5, right: null, left: null }
  }
};


/**
 * In-order BST Traversal (Root, Left, Right). Recursive solution.
 * @param {TreeNode} root
 * @param {number[]} acc
 * @return {number[]}
 */
var inorderTraversal = function(root, acc = []) {
    if (!!root) {
        if (root.left) inorderTraversal(root.left, acc);
        
        acc.push(root.val);
        
        if (root.right) inorderTraversal(root.right, acc);
    }

    return acc;
};

/**
 * In-order BST Traversal (Root, Left, Right). Iterative solution.
 * @param {TreeNode} root
 * @return {number[]}
 */
var inorderTraversal = function(root) {
    /**
    * 1. Create an empty stack S.
    * 2. Initialize current node as root
    * 3. Push the current node to S and set current = current->left until current is NULL
    * 4. If current is NULL and stack is not empty then 
    * 4.1. Pop the top item from stack.
    * 4.2. Print the popped item, set current = popped_item->right 
    * 4.3. Go to step 3.
    * 5. If current is NULL and stack is empty then we are done.
    */
    if (root == null) {
        return [];
    }
    
    const stack = [];
    const result = [];
    
    let isDone = false; 
    let current = root;
    
    while (!isDone) {
        if (current) {   
            stack.push(current);
            current = current.left;
        } else {
            if (stack.length) {
                let currentPopped = stack.pop();
                result.push(currentPopped.val);
                current = currentPopped.right;
            } else {
                isDone = true
            }
        }       
    }
    
    return result;
}