import { rightSideView } from '../../src/tree/binary-tree-right-side-view'

test('二叉树的右视图', () => {
  function TreeNode (val) {
    this.val = val
    this.left = null
    this.right = null
  }

  class Tree {
    constructor () {
      this.root = null
      this.queue = []
      this.insertNum = 0
    }

    insert (val) {
      this.insertNum++ // 插入次数加1
      const node = (!val && typeof val === 'object') ? null : new TreeNode(val) // 判断是否为空节点

      if (!this.root) { // 判断根节点是否存在
        this.root = node // 插入根节点
        node && this.queue.push(this.root) // 非空节点入列
      } else { // 插入非根节点
        const parent = this.queue[0] // 被插入的父节点
        if (!(this.insertNum % 2)) { // 通过插入次数判断左右
          parent.left = node // 插入左边
          node && this.queue.push(parent.left) // 非空节点入列
        } else {
          parent.right = node // 插入右边
          node && this.queue.push(parent.right) // 非空节点入列
          this.queue.shift() // 当前父节点parent 已经不可能再插入子节点，故出列
        }
      }
      return this
    }
  }

  const arr = [1, 2, 3, null, 5, null, 4]

  const tree = new Tree()

  for (const n in arr) {
    tree.insert(arr[n])
  }

  expect(rightSideView(tree.root)).toEqual([1, 3, 4])
})