add: 只出现一次的数字等

test/tree/binary-tree-right-side-view.test.js

@@ -0,0 +1,48 @@
+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])
+})