49 lines
1.4 KiB
JavaScript
49 lines
1.4 KiB
JavaScript
|
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])
|
||
|
|
})
|