mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-08 05:09:08 +08:00
7257641e0e
* Retranslate the complete Japanese online edition from Chinese * Fit Japanese homepage labels and localize heading permalinks * Point Japanese contribution instructions to the Japanese preview * Replace locale conditionals with multilingual UI tables
148 lines
3.7 KiB
JavaScript
148 lines
3.7 KiB
JavaScript
/**
|
||
* File: array_binary_tree.js
|
||
* Created Time: 2023-08-06
|
||
* Author: yuan0221 (yl1452491917@gmail.com)
|
||
*/
|
||
|
||
const { arrToTree } = require('../modules/TreeNode');
|
||
const { printTree } = require('../modules/PrintUtil');
|
||
|
||
/* 配列で表した二分木クラス */
|
||
class ArrayBinaryTree {
|
||
#tree;
|
||
|
||
/* コンストラクタ */
|
||
constructor(arr) {
|
||
this.#tree = arr;
|
||
}
|
||
|
||
/* リストの容量 */
|
||
size() {
|
||
return this.#tree.length;
|
||
}
|
||
|
||
/* インデックス i のノードの値を取得する */
|
||
val(i) {
|
||
// インデックスが範囲外の場合、空きを表す null を返す
|
||
if (i < 0 || i >= this.size()) return null;
|
||
return this.#tree[i];
|
||
}
|
||
|
||
/* インデックス i のノードの左の子ノードのインデックスを取得する */
|
||
left(i) {
|
||
return 2 * i + 1;
|
||
}
|
||
|
||
/* インデックス i のノードの右の子ノードのインデックスを取得する */
|
||
right(i) {
|
||
return 2 * i + 2;
|
||
}
|
||
|
||
/* インデックス i のノードの親ノードのインデックスを取得する */
|
||
parent(i) {
|
||
return Math.floor((i - 1) / 2); // 切り捨て除算
|
||
}
|
||
|
||
/* レベル順走査 */
|
||
levelOrder() {
|
||
let res = [];
|
||
// 配列をそのまま走査する
|
||
for (let i = 0; i < this.size(); i++) {
|
||
if (this.val(i) !== null) res.push(this.val(i));
|
||
}
|
||
return res;
|
||
}
|
||
|
||
/* 深さ優先走査 */
|
||
#dfs(i, order, res) {
|
||
// 空きの場合、戻る
|
||
if (this.val(i) === null) return;
|
||
// 行きがけ順走査
|
||
if (order === 'pre') res.push(this.val(i));
|
||
this.#dfs(this.left(i), order, res);
|
||
// 通りがけ順走査
|
||
if (order === 'in') res.push(this.val(i));
|
||
this.#dfs(this.right(i), order, res);
|
||
// 帰りがけ順走査
|
||
if (order === 'post') res.push(this.val(i));
|
||
}
|
||
|
||
/* 行きがけ順走査 */
|
||
preOrder() {
|
||
const res = [];
|
||
this.#dfs(0, 'pre', res);
|
||
return res;
|
||
}
|
||
|
||
/* 通りがけ順走査 */
|
||
inOrder() {
|
||
const res = [];
|
||
this.#dfs(0, 'in', res);
|
||
return res;
|
||
}
|
||
|
||
/* 帰りがけ順走査 */
|
||
postOrder() {
|
||
const res = [];
|
||
this.#dfs(0, 'post', res);
|
||
return res;
|
||
}
|
||
}
|
||
|
||
/* 実行用コード */
|
||
// 二分木を初期化する
|
||
// ここでは配列から二分木を直接生成する関数を使う
|
||
const arr = Array.of(
|
||
1,
|
||
2,
|
||
3,
|
||
4,
|
||
null,
|
||
6,
|
||
7,
|
||
8,
|
||
9,
|
||
null,
|
||
null,
|
||
12,
|
||
null,
|
||
null,
|
||
15
|
||
);
|
||
|
||
const root = arrToTree(arr);
|
||
console.log('\n二分木を初期化する\n');
|
||
console.log('二分木の配列表現：');
|
||
console.log(arr);
|
||
console.log('二分木の連結リスト表現：');
|
||
printTree(root);
|
||
|
||
// 配列で表した二分木クラス
|
||
const abt = new ArrayBinaryTree(arr);
|
||
|
||
// ノードにアクセスする
|
||
const i = 1;
|
||
const l = abt.left(i);
|
||
const r = abt.right(i);
|
||
const p = abt.parent(i);
|
||
console.log('\n現在のノードのインデックスは ' + i + ' 、値は ' + abt.val(i));
|
||
console.log(
|
||
'左の子ノードのインデックスは ' + l + ' 、値は ' + (l === null ? 'null' : abt.val(l))
|
||
);
|
||
console.log(
|
||
'右の子ノードのインデックスは ' + r + ' 、値は ' + (r === null ? 'null' : abt.val(r))
|
||
);
|
||
console.log(
|
||
'親ノードのインデックスは ' + p + ' 、値は ' + (p === null ? 'null' : abt.val(p))
|
||
);
|
||
|
||
// 木を走査する
|
||
let res = abt.levelOrder();
|
||
console.log('\nレベル順走査：' + res);
|
||
res = abt.preOrder();
|
||
console.log('行きがけ順走査：' + res);
|
||
res = abt.inOrder();
|
||
console.log('通りがけ順走査：' + res);
|
||
res = abt.postOrder();
|
||
console.log('帰りがけ順走査：' + res);
|