如何实现二叉树的深度优先搜索和广度优先搜索

C语言算法：如何实现二叉树的深度优先搜索和广度优先搜索

问题描述

二叉树是一种常用的数据结构，它由一个根节点和两个子树组成，每个子树又是一个二叉树。二叉树的搜索是指按照一定的顺序访问二叉树中的每个节点，通常有两种搜索方式：深度优先搜索和广度优先搜索。深度优先搜索是指从根节点开始，沿着一条路径不断向下访问，直到无法继续，然后回溯到上一个节点，再选择另一条路径继续访问，直到所有节点都被访问过。广度优先搜索是指从根节点开始，按照层次的顺序，依次访问每一层的所有节点，直到所有节点都被访问过。我们如何用C语言编写一个程序，来实现二叉树的深度优先搜索和广度优先搜索呢？

解决方案

要实现二叉树的深度优先搜索和广度优先搜索，我们可以使用递归或者非递归的方法，递归的方法比较简单，非递归的方法需要借助栈或者队列来存储节点。我们分别来看看这两种方法的代码和思路。

递归方法

递归方法的思路是，对于一个二叉树，我们可以先访问根节点，然后递归地搜索左子树和右子树。根据根节点的访问顺序，我们可以分为三种深度优先搜索方式：

• 前序遍历：先访问根节点，再搜索左子树，最后搜索右子树；
• 中序遍历：先搜索左子树，再访问根节点，最后搜索右子树；
• 后序遍历：先搜索左子树，再搜索右子树，最后访问根节点。

下面是用递归的代码示例：

#include <stdio.h>
#include <stdlib.h>

// 定义一个结构体，表示二叉树的节点
typedef struct node
{
    int data; // 节点的数据域
    struct node *left; // 节点的左子树指针
    struct node *right; // 节点的右子树指针
} node;

// 定义一个函数，用于创建一个新的节点
node *create_node(int data)
{
    // 分配内存空间
    node *new_node = (node *)malloc(sizeof(node));
    // 判断是否分配成功
    if (new_node == NULL)
    {
        printf("Memory allocation failed.\n");
        exit(1);
    }
    // 初始化节点的数据域和子树指针
    new_node->data = data;
    new_node->left = NULL;
    new_node->right = NULL;
    // 返回新节点的地址
    return new_node;
}

// 定义一个函数，用于释放一个节点的内存空间
void free_node(node *p)
{
    // 判断节点是否为空
    if (p != NULL)
    {
        // 释放节点的内存空间
        free(p);
    }
}

// 定义一个函数，用于递归地前序遍历二叉树
void preorder_traversal(node *root)
{
    // 判断根节点是否为空
    if (root != NULL)
    {
        // 访问根节点的数据域
        printf("%d ", root->data);
        // 递归地遍历左子树
        preorder_traversal(root->left);
        // 递归地遍历右子树
        preorder_traversal(root->right);
    }
}

// 定义一个函数，用于递归地中序遍历二叉树
void inorder_traversal(node *root)
{
    // 判断根节点是否为空
    if (root != NULL)
    {
        // 递归地遍历左子树
        inorder_traversal(root->left);
        // 访问根节点的数据域
        printf("%d ", root->data);
        // 递归地遍历右子树
        inorder_traversal(root->right);
    }
}

// 定义一个函数，用于递归地后序遍历二叉树
void postorder_traversal(node *root)
{
    // 判断根节点是否为空
    if (root != NULL)
    {
        // 递归地遍历左子树
        postorder_traversal(root->left);
        // 递归地遍历右子树
        postorder_traversal(root->right);
        // 访问根节点的数据域
        printf("%d ", root->data);
    }
}

// 定义一个函数，用于递归地广度优先搜索二叉树
void breadth_first_search(node *root)
{
    // 判断根节点是否为空
    if (root == NULL)
    {
        return;
    }
    // 定义一个队列，用于存储节点的地址
    node *queue[100];
    // 定义两个变量，分别表示队列的头部和尾部的下标
    int front = 0, rear = 0;
    // 将根节点的地址入队
    queue[rear++] = root;
    // 当队列不为空时，循环执行以下操作
    while (front < rear)
    {
        // 将队列头部的节点出队，并访问其数据域
        node *p = queue[front++];
        printf("%d ", p->data);
        // 如果该节点有左子树，将其左子树的地址入队
        if (p->left != NULL)
        {
            queue[rear++] = p->left;
        }
        // 如果该节点有右子树，将其右子树的地址入队
        if (p->right != NULL)
        {
            queue[rear++] = p->right;
        }
    }
}

// 主函数
int main()
{
    // 创建一个示例的二叉树
    //      1
    //     / \
    //    2   3
    //   / \   \
    //  4   5   6
    node *root = create_node(1);
    root->left = create_node(2);
    root->right = create_node(3);
    root->left->left = create_node(4);
    root->left->right = create_node(5);
    root->right->right = create_node(6);

    // 调用函数，分别用四种方式搜索二叉树，并打印结果
    printf("Preorder traversal: ");
    preorder_traversal(root);
    printf("\n");

    printf("Inorder traversal: ");
    inorder_traversal(root);
    printf("\n");

    printf("Postorder traversal: ");
    postorder_traversal(root);
    printf("\n");

    printf("Breadth first search: ");
    breadth_first_search(root);
    printf("\n");

    // 释放二叉树的内存空间
    free_node(root->left->left);
    free_node(root->left->right);
    free_node(root->right->right);
    free_node(root->left);
    free_node(root->right);
    free_node(root);

    // 返回0，表示程序正常结束
    return 0;
}

运行结果：

总结

本文介绍了如何用C语言编写一个程序，来实现二叉树的深度优先搜索和广度优先搜索。我们分别使用了递归和非递归两种方法，给出了代码和解决思路。递归方法是通过定义一个函数，先访问根节点，然后递归地搜索左子树和右子树，根据根节点的访问顺序，可以分为前序遍历、中序遍历和后序