[C语言][剑指offer篇]--从上到下打印二叉树 II(广度优先搜索)

题目描述

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
提示：
节点总数 <= 1000

---

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

理清思路

 广度优先搜索(BFS)：

• BFS基于队列实现。
• 遍历结束的条件：对头(head) = 队尾(tail)
• 当前层的节点个数：head - tail
• 左节点先入队，然后右节点入队
• 函数参数的意义：*returnSize代表返回二叉树的深度，可以当作保存每层节点数据的数组首地址；** returnColumnSizes代表用一维数组返回每层节点的个数。

代码实现

#define MAX_SIZE 10000
int** levelOrder(struct TreeNode* root, int* returnSize, int** returnColumnSizes){

    struct TreeNode *queue[MAX_SIZE];//队列
    int head = 0;//对头
    int tail = 0;//对尾
    int **ans;//返回的数组
    int i = 0;
    int size = 0;//每层的节点个数
    struct TreeNode *curr = NULL;//当前节点

    if(root == NULL)
    {
        *returnSize = 0;
        return NULL;
    }

    ans = (int **)malloc(sizeof(int *)*MAX_SIZE);
    *returnColumnSizes = (int *)malloc(sizeof(int)*MAX_SIZE);

    queue[tail++] = root;
    *returnSize = 0;//代表数的深度

    while(head < tail)
    {
        size = (tail - head + MAX_SIZE) % MAX_SIZE;//防止size超过max_size
        (*returnColumnSizes)[*returnSize] = size;

        ans[*returnSize] = (int *)malloc(sizeof(int)*size);//分配缓存每层节点数据的空间
        
        for(i = 0; i < size; i++)//逐层打印
        {
            curr = queue[head++];//出队
            ans[*returnSize][i] = curr->val;

            if(curr->left)//当前节点的左节点入队
            {
                queue[tail++] = curr->left;
            }
            if(curr->right)//当前节点的右节点入队
            {
                queue[tail++] = curr->right;
            }
        }

        (*returnSize)++;//下一层

    }

    return ans;
}