leetcode算法-创建及遍历二叉树

创建二叉树

二叉树是一种特殊的树，它的子节点个数不超过两个，即左节点和右节点。二叉树的节点包含指向左右节点的指针和节点数据。二叉树可以提供更稳定的访问或查找、插入和删除，相对于链表更快，结点间的连接使用指针，可动态创建节点或释放结点。

在开始刷二叉树的题目前，我们要先创建二叉树。

给定一个数组[1,2,3,4,5,6,7,8,9],请用其生成二叉树。

根据数组创建二叉树算法思路：

• 根据二叉树的性质，编号为index的节点，其左的编号为当前索引加上index再加1，即index2+1，同样的右节点的编号为index2+1。
• 递归计算每个节点的左右节点并赋值。
  代码如下：

class TreeNode {
  constructor(val) {
    this.val = val;
    this.left = null;
    this.right = null;
  }
}

class Tree {
  constructor(list) {
    this.root = null;
    if (list) {
      this.arrayToTree(list, 0);
    }
  }
  /**
   * 根据数组创建二叉树
   * @param {Array} list
   * @param {Number} index
   */
  arrayToTree(list, index) {
    if (!list) {
      return;
    }
    let node = null;
    if (index < list.length && list[index] !== null) {
      node = new TreeNode(list[index]);
      if (!this.root) this.root = node;
      node.left = this.arrayToTree(list, 2 * index + 1);
      node.right = this.arrayToTree(list, 2 * index + 2);
      console.log(index, list[index], node);
    }
    return node;
  }
}

let tree = new Tree([3, 9, 20, null, null, 15, 7]);
console.log(tree);

二叉树的遍历

创建二叉树后，就可以对二叉树进行遍历。二叉树常见的遍历方法有层序遍历、前序遍历、中序遍历、后序遍历。

层序遍历

leetcode102：

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。
示例： 二叉树：[3,9,20,null,null,15,7]
返回其层序遍历结果：

[ [3], [9,20], [15,7] ]

层序遍历的思路：

• 首先根元素入队
• 当队列不为空的时候
  • 求当前队列的长度 size_i,依次从队列中取 size_i个元素,这些元素即为第i层所有元素
  • 遍历 0至size_i的元素，队列首位元素出列，将其左右节点添加都队列中作为第i+1层元素
  • 进入下一次迭代
    代码如下：

function levelOrder(root) {
  if (!root) return [];
  let result = [];
  let queen = [root];
  let k = 0;
  while (queen.length) {
    let size = queen.length;
    result[k] = [];
    for (let i = 0; i < size; i++) {
      let curr = queen.shift();
      result[k].push(curr.val);
      if (curr.left) queen.push(curr.left);
      if (curr.right) queen.push(curr.right);
    }
    k++;
  }
  return result;
}

前序遍历

二叉树前序遍历的定义：

（1）访问根节点
（2） 前序遍历左子树
（3） 前序遍历右子树

例子：

前序遍历二叉树[3, 9, 20, null, null, 15, 7],依次输出输出3,9,20,15,7

递归前序遍历

// visit 对数据元素进行操作的函数
function preOrderTraverse(root,visit){
    if(!root) return;
    visit && visit(root.val);
    preOrderTraverse(root.left,visit);
    preOrderTraverse(root.right,visit);
}
let tree = new Tree([3, 9, 20, null, null, 15, 7]);
preOrderTraverse(tree.root,console.log);

非递归前序遍历

function preOrderTraverse(root, visit) {
  if (!root) return;
  let stack = [];
  stack.push(root)
  let current;
  while (stack.length) {
   // 访问栈顶元素
   current=stack.pop(); 
   visit && visit(current.val);
   if(current.right) stack.push(current.right);
   if(current.left) stack.push(current.left);
  }
}

中序遍历

二叉树中序遍历的定义：

（1）中序遍历左子树
（2）访问根节点
（3）中序遍历右子树

例子：

中序遍历二叉树[3, 9, 20, null, null, 15, 7],依次输出输出9,3,15,20,7

递归中序遍历

let tree = new Tree([3, 9, 20, null, null, 15, 7]);
// visit 对数据元素进行操作的函数
function inOrderTraverse(root,visit){
    if(!root) return;
    inOrderTraverse(root.left,visit);
    visit && visit(root.val);
    inOrderTraverse(root.right,visit);
}
inOrderTraverse(tree.root,console.log);

非递归中序遍历

function inOrderTraverse(root, visit) {
  if (!root) return;
  let stack = [];
  let current=root;
  while (current || stack.length) {
   //  遍历左子树
    while (current) {
      stack.push(current);
      current=current.left
    }
    // 访问栈顶元素
    current = stack.pop();
    visit && visit(current.val);
    // 遍历右子树
    current=current.right
  }
}

后序遍历

二叉树后序遍历的定义：

（1）后序遍历左子树
（2）后序遍历右子树
（3）访问根节点
例子：
后序遍历二叉树[3, 9, 20, null, null, 15, 7],依次输出输出9,15,7,20,3

递归后序遍历

let tree = new Tree([3, 9, 20, null, null, 15, 7]);
// visit 对数据元素进行操作的函数
function posterOrderTraverse(root,visit){
    if(!root) return;
    posterOrderTraverse(root.left,visit);
    posterOrderTraverse(root.right,visit);
    visit && visit(root.val);
}
posterOrderTraverse(tree.root,console.log);