数据结构：二叉树初阶与链式实现

重要性质： 对于具有 n 个结点的完全二叉树，如果按照从上至下从左至右的数组顺序对所有结点从 0 开始编号：

• 若父亲在数组中下标为 i，则该父亲的左孩子下标为 2 * i + 1；右孩子下标为 2 * i + 2。
• 若孩子在数组中下标为 i，则该孩子的父亲下标为 (i - 1) / 2。 注意：前提是这些下标所对应的值存在。

三、二叉树的实现

1. 链式存储

前面提到完全二叉树可以用数组存储，而非完全二叉树若强行用数组会浪费大量空间。此时我们使用链式存储。

链式存储即用链表来表示一颗二叉树，用链来指示元素的逻辑关系。通常的方法创建一链表，使每个结点由三个域组成：数据域和左右指针域。左右指针分别用来给出该结点左孩子和右孩子所在结点的地址。

2. 节点定义

与单链表类似，二叉树中的每一个节点有以下三个成员：

1. 节点中存储的数据
2. 指向节点左孩子的指针
3. 指向节点右孩子的指针

为了便于类型修改，我们使用 typedef 重定义数据类型。

// BTNode.h
#pragma once
#include "Queue.h"

typedef char BTDataType;

typedef struct BinaryTreeNode {
    BTDataType data;
    struct BinaryTreeNode* left;
    struct BinaryTreeNode* right;
}BTNode;

3. 构建二叉树

这里通过一个前序遍历的数组 "ABD##E#H##CF##G##" 来构建二叉树。'#' 代表空节点。

// BTNode.c
BTNode* BinaryTreeCreate(BTDataType* ch, int* pi) {
    if (ch[*pi] == '#') {
        return NULL;
    }
    BTNode* newnode = (BTNode*)malloc(sizeof(BTNode));
    if (newnode == NULL) {
        perror("malloc fail");
        return NULL;
    }
    newnode->data = ch[(*pi)++];
    newnode->left = BinaryTreeCreate(ch, pi);
    (*pi)++;
    newnode->right = BinaryTreeCreate(ch, pi);
    return newnode;
}

4. 二叉树遍历

前序、中序、后序遍历

这三种遍历主要在于访问根节点的时机不同，均采用递归实现。

• 前序遍历：根 -> 左 -> 右
• 中序遍历：左 -> 根 -> 右
• 后序遍历：左 -> 右 -> 根

// 前序遍历
void BinaryTreePrevOrder(BTNode* root) {
    if (root == NULL) {
        return;
    }
    printf("%c ", root->data);
    BinaryTreePrevOrder(root->left);
    BinaryTreePrevOrder(root->right);
}

// 中序遍历
void BinaryTreeInOrder(BTNode* root) {
    if (root == NULL) {
        return;
    }
    BinaryTreeInOrder(root->left);
    printf("%c ", root->data);
    BinaryTreeInOrder(root->right);
}

// 后序遍历
void BinaryTreePostOrder(BTNode* root) {
    if (root == NULL) {
        return;
    }
    BinaryTreePostOrder(root->left);
    BinaryTreePostOrder(root->right);
    printf("%c ", root->data);
}

层序遍历

层序遍历需要借助队列来实现。思路是每次取出队头结点，将其左右孩子入队，直到队列为空。

// 层序遍历
void BinaryTreeLevelOrder(BTNode* root) {
    Queue Q;
    QueueInit(&Q);
    if (root) {
        QueuePush(&Q, root);
    }
    while (!QueueEmpty(&Q)) {
        BTNode* front = QueueFront(&Q);
        printf("%c ", front->data);
        QueuePop(&Q);
        if (front->left) {
            QueuePush(&Q, front->left);
        }
        if (front->right) {
            QueuePush(&Q, front->right);
        }
    }
    QueueDestroy(&Q);
}

5. 其他常用操作

节点个数计算

利用递归，根节点为空返回 0，否则返回左子树 + 右子树 + 1。

int BinaryTreeSize(BTNode* root) {
    return root == NULL ? 0 : BinaryTreeSize(root->left) + BinaryTreeSize(root->right) + 1;
}

二叉树高度计算

返回左右子树高度较高的一边并 + 1。

int BinaryTreeHight(BTNode* root) {
    if (root == NULL) {
        return 0;
    }
    int leftH = BinaryTreeHight(root->left);
    int rightH = BinaryTreeHight(root->right);
    return (leftH > rightH ? leftH : rightH) + 1;
}

第 k 层节点个数

初始 k，若该层不是目标层次，就 k - 1 并递归左右子树。到达第 k 层返回 1。

int BinaryTreeLevelKSize(BTNode* root, int k) {
    if (root == NULL) {
        return 0;
    }
    if (k == 1) {
        return 1;
    }
    return BinaryTreeLevelKSize(root->left, k - 1) + BinaryTreeLevelKSize(root->right, k - 1);
}

查找值为 x 的节点

递归查找，先判断根节点，再找左子树，若未找到再找右子树。

BTNode* BinaryTreeFind(BTNode* root, BTDataType x) {
    if (root == NULL) {
        return NULL;
    }
    if (root->data == x) {
        return root;
    }
    BTNode* ret1 = BinaryTreeFind(root->left, x);
    if (ret1) {
        return ret1;
    }
    BTNode* ret2 = BinaryTreeFind(root->right, x);
    if (ret2) {
        return ret2;
    }
    return NULL;
}

判断是否是完全二叉树

利用层序遍历，遇到空节点后，后续不应再有非空节点。

int BinaryTreeComplete(BTNode* root) {
    Queue Q;
    QueueInit(&Q);
    if (root) {
        QueuePush(&Q, root);
    }
    while (!QueueEmpty(&Q)) {
        BTNode* front = QueueFront(&Q);
        QueuePop(&Q);
        if (front == NULL) {
            break;
        }
        QueuePush(&Q, front->left);
        QueuePush(&Q, front->right);
    }
    while (!QueueEmpty(&Q)) {
        BTNode* front = QueueFront(&Q);
        QueuePop(&Q);
        if (front != NULL) {
            QueueDestroy(&Q);
            return 0;
        }
    }
    QueueDestroy(&Q);
    return 1;
}

四、完整代码结构

1. 队列实现 (Queue.h / Queue.c)

// Queue.h
#pragma once
#define _CRT_SECURE_NO_WARNINGS 1
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <assert.h>

typedef int QDataType;

typedef struct QListNode {
    QDataType data;
    struct QListNode* next;
}QNode;

typedef struct Queue {
    QNode* front;
    QNode* tail;
    int size;
}Queue;

void QueueInit(Queue* q);
void QueuePush(Queue* q, QDataType data);
void QueuePop(Queue* q);
QDataType QueueFront(Queue* q);
QDataType QueueBack(Queue* q);
int QueueSize(Queue* q);
int QueueEmpty(Queue* q);
void QueueDestroy(Queue* q);

// Queue.c
#include "Queue.h"

void QueueInit(Queue* q) {
    assert(q);
    q->front = NULL;
    q->tail = NULL;
    q->size = 0;
}

void QueuePush(Queue* q, QDataType data) {
    assert(q);
    QNode* tmp = (QNode*)malloc(sizeof(QNode));
    if (tmp == NULL) {
        perror("malloc");
        return;
    }
    tmp->data = data;
    tmp->next = NULL;
    if (q->size == 0) {
        q->front = tmp;
        q->tail = tmp;
    } else {
        q->tail->next = tmp;
        q->tail = tmp;
    }
    q->size++;
}

void QueuePop(Queue* q) {
    assert(q);
    assert(q->size > 0);
    if (q->size == 1) {
        free(q->front);
        q->front = NULL;
        q->tail = NULL;
    } else {
        QNode* del = q->front;
        q->front = q->front->next;
        free(del);
    }
    q->size--;
}

QDataType QueueFront(Queue* q) {
    assert(q);
    assert(q->size > 0);
    return q->front->data;
}

QDataType QueueBack(Queue* q) {
    assert(q);
    assert(q->size > 0);
    return q->tail->data;
}

int QueueSize(Queue* q) {
    assert(q);
    return q->size;
}

int QueueEmpty(Queue* q) {
    assert(q);
    return q->size == 0;
}

void QueueDestroy(Queue* q) {
    assert(q);
    QNode* cur = q->front;
    while (cur) {
        QNode* next = cur->next;
        free(cur);
        cur = next;
    }
    q->front = q->tail = NULL;
    q->size = 0;
}

2. 测试文件 (Test.c)

#include "BTNode.h"

int main() {
    char ch[100] = "ABD##E#H##CF##G##";
    int i = 0;
    BTNode* T = BinaryTreeCreate(ch, &i);

    printf("节点个数：%d\n", BinaryTreeSize(T));
    printf("叶子节点个数：%d\n", BinaryTreeLeafSize(T));
    printf("高度：%d\n", BinaryTreeHight(T));
    printf("第 3 层节点个数：%d\n", BinaryTreeLevelKSize(T, 3));

    BTNode* ret = BinaryTreeFind(T, 'G');
    if (ret) {
        printf("找到 G: %c\n", ret->data);
    }

    printf("是否完全二叉树：%s\n", BinaryTreeComplete(T) ? "是" : "否");

    printf("前序：");
    BinaryTreePrevOrder(T);
    printf("\n");
    printf("中序：");
    BinaryTreeInOrder(T);
    printf("\n");
    printf("后序：");
    BinaryTreePostOrder(T);
    printf("\n");
    printf("层序：");
    BinaryTreeLevelOrder(T);
    printf("\n");

    BinaryTreeDestory(&T);
    return 0;
}

本文涵盖了二叉树的核心概念与工程实现细节。通过上述代码，你可以直接编译运行并观察不同遍历方式及统计函数的效果。在实际开发中，理解这些底层逻辑有助于更好地设计算法和处理复杂的数据关系。