---
comments: true
---
# 7.3 二元樹陣列表示
在鏈結串列表示下,二元樹的儲存單元為節點 `TreeNode` ,節點之間透過指標相連線。上一節介紹了鏈結串列表示下的二元樹的各項基本操作。
那麼,我們能否用陣列來表示二元樹呢?答案是肯定的。
## 7.3.1 表示完美二元樹
先分析一個簡單案例。給定一棵完美二元樹,我們將所有節點按照層序走訪的順序儲存在一個陣列中,則每個節點都對應唯一的陣列索引。
根據層序走訪的特性,我們可以推導出父節點索引與子節點索引之間的“對映公式”:**若某節點的索引為 $i$ ,則該節點的左子節點索引為 $2i + 1$ ,右子節點索引為 $2i + 2$** 。圖 7-12 展示了各個節點索引之間的對映關係。
![完美二元樹的陣列表示](array_representation_of_tree.assets/array_representation_binary_tree.png){ class="animation-figure" }
圖 7-12 完美二元樹的陣列表示
**對映公式的角色相當於鏈結串列中的節點引用(指標)**。給定陣列中的任意一個節點,我們都可以透過對映公式來訪問它的左(右)子節點。
## 7.3.2 表示任意二元樹
完美二元樹是一個特例,在二元樹的中間層通常存在許多 `None` 。由於層序走訪序列並不包含這些 `None` ,因此我們無法僅憑該序列來推測 `None` 的數量和分佈位置。**這意味著存在多種二元樹結構都符合該層序走訪序列**。
如圖 7-13 所示,給定一棵非完美二元樹,上述陣列表示方法已經失效。
![層序走訪序列對應多種二元樹可能性](array_representation_of_tree.assets/array_representation_without_empty.png){ class="animation-figure" }
圖 7-13 層序走訪序列對應多種二元樹可能性
為了解決此問題,**我們可以考慮在層序走訪序列中顯式地寫出所有 `None`** 。如圖 7-14 所示,這樣處理後,層序走訪序列就可以唯一表示二元樹了。示例程式碼如下:
=== "Python"
```python title=""
# 二元樹的陣列表示
# 使用 None 來表示空位
tree = [1, 2, 3, 4, None, 6, 7, 8, 9, None, None, 12, None, None, 15]
```
=== "C++"
```cpp title=""
/* 二元樹的陣列表示 */
// 使用 int 最大值 INT_MAX 標記空位
vector tree = {1, 2, 3, 4, INT_MAX, 6, 7, 8, 9, INT_MAX, INT_MAX, 12, INT_MAX, INT_MAX, 15};
```
=== "Java"
```java title=""
/* 二元樹的陣列表示 */
// 使用 int 的包裝類別 Integer ,就可以使用 null 來標記空位
Integer[] tree = { 1, 2, 3, 4, null, 6, 7, 8, 9, null, null, 12, null, null, 15 };
```
=== "C#"
```csharp title=""
/* 二元樹的陣列表示 */
// 使用 int? 可空型別 ,就可以使用 null 來標記空位
int?[] tree = [1, 2, 3, 4, null, 6, 7, 8, 9, null, null, 12, null, null, 15];
```
=== "Go"
```go title=""
/* 二元樹的陣列表示 */
// 使用 any 型別的切片, 就可以使用 nil 來標記空位
tree := []any{1, 2, 3, 4, nil, 6, 7, 8, 9, nil, nil, 12, nil, nil, 15}
```
=== "Swift"
```swift title=""
/* 二元樹的陣列表示 */
// 使用 Int? 可空型別 ,就可以使用 nil 來標記空位
let tree: [Int?] = [1, 2, 3, 4, nil, 6, 7, 8, 9, nil, nil, 12, nil, nil, 15]
```
=== "JS"
```javascript title=""
/* 二元樹的陣列表示 */
// 使用 null 來表示空位
let tree = [1, 2, 3, 4, null, 6, 7, 8, 9, null, null, 12, null, null, 15];
```
=== "TS"
```typescript title=""
/* 二元樹的陣列表示 */
// 使用 null 來表示空位
let tree: (number | null)[] = [1, 2, 3, 4, null, 6, 7, 8, 9, null, null, 12, null, null, 15];
```
=== "Dart"
```dart title=""
/* 二元樹的陣列表示 */
// 使用 int? 可空型別 ,就可以使用 null 來標記空位
List tree = [1, 2, 3, 4, null, 6, 7, 8, 9, null, null, 12, null, null, 15];
```
=== "Rust"
```rust title=""
/* 二元樹的陣列表示 */
// 使用 None 來標記空位
let tree = [Some(1), Some(2), Some(3), Some(4), None, Some(6), Some(7), Some(8), Some(9), None, None, Some(12), None, None, Some(15)];
```
=== "C"
```c title=""
/* 二元樹的陣列表示 */
// 使用 int 最大值標記空位,因此要求節點值不能為 INT_MAX
int tree[] = {1, 2, 3, 4, INT_MAX, 6, 7, 8, 9, INT_MAX, INT_MAX, 12, INT_MAX, INT_MAX, 15};
```
=== "Kotlin"
```kotlin title=""
/* 二元樹的陣列表示 */
// 使用 null 來表示空位
val tree = arrayOf( 1, 2, 3, 4, null, 6, 7, 8, 9, null, null, 12, null, null, 15 )
```
=== "Ruby"
```ruby title=""
```
=== "Zig"
```zig title=""
```
![任意型別二元樹的陣列表示](array_representation_of_tree.assets/array_representation_with_empty.png){ class="animation-figure" }
圖 7-14 任意型別二元樹的陣列表示
值得說明的是,**完全二元樹非常適合使用陣列來表示**。回顧完全二元樹的定義,`None` 只出現在最底層且靠右的位置,**因此所有 `None` 一定出現在層序走訪序列的末尾**。
這意味著使用陣列表示完全二元樹時,可以省略儲存所有 `None` ,非常方便。圖 7-15 給出了一個例子。
![完全二元樹的陣列表示](array_representation_of_tree.assets/array_representation_complete_binary_tree.png){ class="animation-figure" }
圖 7-15 完全二元樹的陣列表示
以下程式碼實現了一棵基於陣列表示的二元樹,包括以下幾種操作。
- 給定某節點,獲取它的值、左(右)子節點、父節點。
- 獲取前序走訪、中序走訪、後序走訪、層序走訪序列。
=== "Python"
```python title="array_binary_tree.py"
class ArrayBinaryTree:
"""陣列表示下的二元樹類別"""
def __init__(self, arr: list[int | None]):
"""建構子"""
self._tree = list(arr)
def size(self):
"""串列容量"""
return len(self._tree)
def val(self, i: int) -> int:
"""獲取索引為 i 節點的值"""
# 若索引越界,則返回 None ,代表空位
if i < 0 or i >= self.size():
return None
return self._tree[i]
def left(self, i: int) -> int | None:
"""獲取索引為 i 節點的左子節點的索引"""
return 2 * i + 1
def right(self, i: int) -> int | None:
"""獲取索引為 i 節點的右子節點的索引"""
return 2 * i + 2
def parent(self, i: int) -> int | None:
"""獲取索引為 i 節點的父節點的索引"""
return (i - 1) // 2
def level_order(self) -> list[int]:
"""層序走訪"""
self.res = []
# 直接走訪陣列
for i in range(self.size()):
if self.val(i) is not None:
self.res.append(self.val(i))
return self.res
def dfs(self, i: int, order: str):
"""深度優先走訪"""
if self.val(i) is None:
return
# 前序走訪
if order == "pre":
self.res.append(self.val(i))
self.dfs(self.left(i), order)
# 中序走訪
if order == "in":
self.res.append(self.val(i))
self.dfs(self.right(i), order)
# 後序走訪
if order == "post":
self.res.append(self.val(i))
def pre_order(self) -> list[int]:
"""前序走訪"""
self.res = []
self.dfs(0, order="pre")
return self.res
def in_order(self) -> list[int]:
"""中序走訪"""
self.res = []
self.dfs(0, order="in")
return self.res
def post_order(self) -> list[int]:
"""後序走訪"""
self.res = []
self.dfs(0, order="post")
return self.res
```
=== "C++"
```cpp title="array_binary_tree.cpp"
/* 陣列表示下的二元樹類別 */
class ArrayBinaryTree {
public:
/* 建構子 */
ArrayBinaryTree(vector arr) {
tree = arr;
}
/* 串列容量 */
int size() {
return tree.size();
}
/* 獲取索引為 i 節點的值 */
int val(int i) {
// 若索引越界,則返回 INT_MAX ,代表空位
if (i < 0 || i >= size())
return INT_MAX;
return tree[i];
}
/* 獲取索引為 i 節點的左子節點的索引 */
int left(int i) {
return 2 * i + 1;
}
/* 獲取索引為 i 節點的右子節點的索引 */
int right(int i) {
return 2 * i + 2;
}
/* 獲取索引為 i 節點的父節點的索引 */
int parent(int i) {
return (i - 1) / 2;
}
/* 層序走訪 */
vector levelOrder() {
vector res;
// 直接走訪陣列
for (int i = 0; i < size(); i++) {
if (val(i) != INT_MAX)
res.push_back(val(i));
}
return res;
}
/* 前序走訪 */
vector preOrder() {
vector res;
dfs(0, "pre", res);
return res;
}
/* 中序走訪 */
vector inOrder() {
vector res;
dfs(0, "in", res);
return res;
}
/* 後序走訪 */
vector postOrder() {
vector res;
dfs(0, "post", res);
return res;
}
private:
vector tree;
/* 深度優先走訪 */
void dfs(int i, string order, vector &res) {
// 若為空位,則返回
if (val(i) == INT_MAX)
return;
// 前序走訪
if (order == "pre")
res.push_back(val(i));
dfs(left(i), order, res);
// 中序走訪
if (order == "in")
res.push_back(val(i));
dfs(right(i), order, res);
// 後序走訪
if (order == "post")
res.push_back(val(i));
}
};
```
=== "Java"
```java title="array_binary_tree.java"
/* 陣列表示下的二元樹類別 */
class ArrayBinaryTree {
private List tree;
/* 建構子 */
public ArrayBinaryTree(List arr) {
tree = new ArrayList<>(arr);
}
/* 串列容量 */
public int size() {
return tree.size();
}
/* 獲取索引為 i 節點的值 */
public Integer val(int i) {
// 若索引越界,則返回 null ,代表空位
if (i < 0 || i >= size())
return null;
return tree.get(i);
}
/* 獲取索引為 i 節點的左子節點的索引 */
public Integer left(int i) {
return 2 * i + 1;
}
/* 獲取索引為 i 節點的右子節點的索引 */
public Integer right(int i) {
return 2 * i + 2;
}
/* 獲取索引為 i 節點的父節點的索引 */
public Integer parent(int i) {
return (i - 1) / 2;
}
/* 層序走訪 */
public List levelOrder() {
List res = new ArrayList<>();
// 直接走訪陣列
for (int i = 0; i < size(); i++) {
if (val(i) != null)
res.add(val(i));
}
return res;
}
/* 深度優先走訪 */
private void dfs(Integer i, String order, List res) {
// 若為空位,則返回
if (val(i) == null)
return;
// 前序走訪
if ("pre".equals(order))
res.add(val(i));
dfs(left(i), order, res);
// 中序走訪
if ("in".equals(order))
res.add(val(i));
dfs(right(i), order, res);
// 後序走訪
if ("post".equals(order))
res.add(val(i));
}
/* 前序走訪 */
public List preOrder() {
List res = new ArrayList<>();
dfs(0, "pre", res);
return res;
}
/* 中序走訪 */
public List inOrder() {
List res = new ArrayList<>();
dfs(0, "in", res);
return res;
}
/* 後序走訪 */
public List postOrder() {
List res = new ArrayList<>();
dfs(0, "post", res);
return res;
}
}
```
=== "C#"
```csharp title="array_binary_tree.cs"
/* 陣列表示下的二元樹類別 */
class ArrayBinaryTree(List arr) {
List tree = new(arr);
/* 串列容量 */
public int Size() {
return tree.Count;
}
/* 獲取索引為 i 節點的值 */
public int? Val(int i) {
// 若索引越界,則返回 null ,代表空位
if (i < 0 || i >= Size())
return null;
return tree[i];
}
/* 獲取索引為 i 節點的左子節點的索引 */
public int Left(int i) {
return 2 * i + 1;
}
/* 獲取索引為 i 節點的右子節點的索引 */
public int Right(int i) {
return 2 * i + 2;
}
/* 獲取索引為 i 節點的父節點的索引 */
public int Parent(int i) {
return (i - 1) / 2;
}
/* 層序走訪 */
public List LevelOrder() {
List res = [];
// 直接走訪陣列
for (int i = 0; i < Size(); i++) {
if (Val(i).HasValue)
res.Add(Val(i)!.Value);
}
return res;
}
/* 深度優先走訪 */
void DFS(int i, string order, List res) {
// 若為空位,則返回
if (!Val(i).HasValue)
return;
// 前序走訪
if (order == "pre")
res.Add(Val(i)!.Value);
DFS(Left(i), order, res);
// 中序走訪
if (order == "in")
res.Add(Val(i)!.Value);
DFS(Right(i), order, res);
// 後序走訪
if (order == "post")
res.Add(Val(i)!.Value);
}
/* 前序走訪 */
public List PreOrder() {
List res = [];
DFS(0, "pre", res);
return res;
}
/* 中序走訪 */
public List InOrder() {
List res = [];
DFS(0, "in", res);
return res;
}
/* 後序走訪 */
public List PostOrder() {
List res = [];
DFS(0, "post", res);
return res;
}
}
```
=== "Go"
```go title="array_binary_tree.go"
/* 陣列表示下的二元樹類別 */
type arrayBinaryTree struct {
tree []any
}
/* 建構子 */
func newArrayBinaryTree(arr []any) *arrayBinaryTree {
return &arrayBinaryTree{
tree: arr,
}
}
/* 串列容量 */
func (abt *arrayBinaryTree) size() int {
return len(abt.tree)
}
/* 獲取索引為 i 節點的值 */
func (abt *arrayBinaryTree) val(i int) any {
// 若索引越界,則返回 null ,代表空位
if i < 0 || i >= abt.size() {
return nil
}
return abt.tree[i]
}
/* 獲取索引為 i 節點的左子節點的索引 */
func (abt *arrayBinaryTree) left(i int) int {
return 2*i + 1
}
/* 獲取索引為 i 節點的右子節點的索引 */
func (abt *arrayBinaryTree) right(i int) int {
return 2*i + 2
}
/* 獲取索引為 i 節點的父節點的索引 */
func (abt *arrayBinaryTree) parent(i int) int {
return (i - 1) / 2
}
/* 層序走訪 */
func (abt *arrayBinaryTree) levelOrder() []any {
var res []any
// 直接走訪陣列
for i := 0; i < abt.size(); i++ {
if abt.val(i) != nil {
res = append(res, abt.val(i))
}
}
return res
}
/* 深度優先走訪 */
func (abt *arrayBinaryTree) dfs(i int, order string, res *[]any) {
// 若為空位,則返回
if abt.val(i) == nil {
return
}
// 前序走訪
if order == "pre" {
*res = append(*res, abt.val(i))
}
abt.dfs(abt.left(i), order, res)
// 中序走訪
if order == "in" {
*res = append(*res, abt.val(i))
}
abt.dfs(abt.right(i), order, res)
// 後序走訪
if order == "post" {
*res = append(*res, abt.val(i))
}
}
/* 前序走訪 */
func (abt *arrayBinaryTree) preOrder() []any {
var res []any
abt.dfs(0, "pre", &res)
return res
}
/* 中序走訪 */
func (abt *arrayBinaryTree) inOrder() []any {
var res []any
abt.dfs(0, "in", &res)
return res
}
/* 後序走訪 */
func (abt *arrayBinaryTree) postOrder() []any {
var res []any
abt.dfs(0, "post", &res)
return res
}
```
=== "Swift"
```swift title="array_binary_tree.swift"
/* 陣列表示下的二元樹類別 */
class ArrayBinaryTree {
private var tree: [Int?]
/* 建構子 */
init(arr: [Int?]) {
tree = arr
}
/* 串列容量 */
func size() -> Int {
tree.count
}
/* 獲取索引為 i 節點的值 */
func val(i: Int) -> Int? {
// 若索引越界,則返回 null ,代表空位
if i < 0 || i >= size() {
return nil
}
return tree[i]
}
/* 獲取索引為 i 節點的左子節點的索引 */
func left(i: Int) -> Int {
2 * i + 1
}
/* 獲取索引為 i 節點的右子節點的索引 */
func right(i: Int) -> Int {
2 * i + 2
}
/* 獲取索引為 i 節點的父節點的索引 */
func parent(i: Int) -> Int {
(i - 1) / 2
}
/* 層序走訪 */
func levelOrder() -> [Int] {
var res: [Int] = []
// 直接走訪陣列
for i in 0 ..< size() {
if let val = val(i: i) {
res.append(val)
}
}
return res
}
/* 深度優先走訪 */
private func dfs(i: Int, order: String, res: inout [Int]) {
// 若為空位,則返回
guard let val = val(i: i) else {
return
}
// 前序走訪
if order == "pre" {
res.append(val)
}
dfs(i: left(i: i), order: order, res: &res)
// 中序走訪
if order == "in" {
res.append(val)
}
dfs(i: right(i: i), order: order, res: &res)
// 後序走訪
if order == "post" {
res.append(val)
}
}
/* 前序走訪 */
func preOrder() -> [Int] {
var res: [Int] = []
dfs(i: 0, order: "pre", res: &res)
return res
}
/* 中序走訪 */
func inOrder() -> [Int] {
var res: [Int] = []
dfs(i: 0, order: "in", res: &res)
return res
}
/* 後序走訪 */
func postOrder() -> [Int] {
var res: [Int] = []
dfs(i: 0, order: "post", res: &res)
return res
}
}
```
=== "JS"
```javascript title="array_binary_tree.js"
/* 陣列表示下的二元樹類別 */
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;
}
}
```
=== "TS"
```typescript title="array_binary_tree.ts"
/* 陣列表示下的二元樹類別 */
class ArrayBinaryTree {
#tree: (number | null)[];
/* 建構子 */
constructor(arr: (number | null)[]) {
this.#tree = arr;
}
/* 串列容量 */
size(): number {
return this.#tree.length;
}
/* 獲取索引為 i 節點的值 */
val(i: number): number | null {
// 若索引越界,則返回 null ,代表空位
if (i < 0 || i >= this.size()) return null;
return this.#tree[i];
}
/* 獲取索引為 i 節點的左子節點的索引 */
left(i: number): number {
return 2 * i + 1;
}
/* 獲取索引為 i 節點的右子節點的索引 */
right(i: number): number {
return 2 * i + 2;
}
/* 獲取索引為 i 節點的父節點的索引 */
parent(i: number): number {
return Math.floor((i - 1) / 2); // 向下整除
}
/* 層序走訪 */
levelOrder(): number[] {
let res = [];
// 直接走訪陣列
for (let i = 0; i < this.size(); i++) {
if (this.val(i) !== null) res.push(this.val(i));
}
return res;
}
/* 深度優先走訪 */
#dfs(i: number, order: Order, res: (number | null)[]): void {
// 若為空位,則返回
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(): (number | null)[] {
const res = [];
this.#dfs(0, 'pre', res);
return res;
}
/* 中序走訪 */
inOrder(): (number | null)[] {
const res = [];
this.#dfs(0, 'in', res);
return res;
}
/* 後序走訪 */
postOrder(): (number | null)[] {
const res = [];
this.#dfs(0, 'post', res);
return res;
}
}
```
=== "Dart"
```dart title="array_binary_tree.dart"
/* 陣列表示下的二元樹類別 */
class ArrayBinaryTree {
late List _tree;
/* 建構子 */
ArrayBinaryTree(this._tree);
/* 串列容量 */
int size() {
return _tree.length;
}
/* 獲取索引為 i 節點的值 */
int? val(int i) {
// 若索引越界,則返回 null ,代表空位
if (i < 0 || i >= size()) {
return null;
}
return _tree[i];
}
/* 獲取索引為 i 節點的左子節點的索引 */
int? left(int i) {
return 2 * i + 1;
}
/* 獲取索引為 i 節點的右子節點的索引 */
int? right(int i) {
return 2 * i + 2;
}
/* 獲取索引為 i 節點的父節點的索引 */
int? parent(int i) {
return (i - 1) ~/ 2;
}
/* 層序走訪 */
List levelOrder() {
List res = [];
for (int i = 0; i < size(); i++) {
if (val(i) != null) {
res.add(val(i)!);
}
}
return res;
}
/* 深度優先走訪 */
void dfs(int i, String order, List res) {
// 若為空位,則返回
if (val(i) == null) {
return;
}
// 前序走訪
if (order == 'pre') {
res.add(val(i));
}
dfs(left(i)!, order, res);
// 中序走訪
if (order == 'in') {
res.add(val(i));
}
dfs(right(i)!, order, res);
// 後序走訪
if (order == 'post') {
res.add(val(i));
}
}
/* 前序走訪 */
List preOrder() {
List res = [];
dfs(0, 'pre', res);
return res;
}
/* 中序走訪 */
List inOrder() {
List res = [];
dfs(0, 'in', res);
return res;
}
/* 後序走訪 */
List postOrder() {
List res = [];
dfs(0, 'post', res);
return res;
}
}
```
=== "Rust"
```rust title="array_binary_tree.rs"
/* 陣列表示下的二元樹類別 */
struct ArrayBinaryTree {
tree: Vec