二叉树-BM26 求二叉树的层序遍历

最新推荐文章于 2024-10-16 19:08:56 发布

jiqian4953

最新推荐文章于 2024-10-16 19:08:56 发布

阅读量84

点赞数

分类专栏：刷题文章标签：数据结构算法

本文链接：https://blog.csdn.net/jiqian4953/article/details/129017910

版权

刷题专栏收录该内容

34 篇文章 0 订阅

订阅专栏

该代码实现了二叉树的层序遍历，利用队列数据结构，按层次从左到右返回节点值。首先检查根节点是否为空，然后初始化队列和变量，将根节点入队。在循环中，每次处理一层的节点，将当前层的节点值存储到结果数组中，并将子节点入队，更新列宽和返回大小。

摘要由CSDN通过智能技术生成

BM25 二叉树的后序遍历

给定一个二叉树，返回该二叉树层序遍历的结果，（从左到右，一层一层地遍历）

解法(队列)

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 * };
 */
/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 *
 * @param root TreeNode类
 * @return int整型二维数组
 * @return int* returnSize 返回数组行数
 * @return int** returnColumnSizes 返回数组列数
 */
typedef struct TreeNode* pnode;


int** levelOrder(struct TreeNode* root, int* returnSize,  int** returnColumnSizes ) {
    // write code here
    if (root == NULL) {
        return NULL;
    }

    pnode queue[1500];
    int front = 0;
    int rear = 0;
    queue[rear++] = root;

    int** ans = (int**)malloc(1500 * sizeof(int*));
    *returnSize = 0;
    *returnColumnSizes = (int*)malloc(1500 * sizeof(int));

    while (front != rear) {
        int prerear = rear;
        int k = 0;
        ans[*returnSize] = (int*)malloc((prerear - front) * sizeof(int));
        while (front < prerear) {
            pnode p = queue[front++];
            ans[*returnSize][k++] = p->val;
            if (p->left != NULL) {
                queue[rear++] = p->left;
            }
            if (p->right != NULL) {
                queue[rear++] = p->right;
            }
        }
        (*returnColumnSizes)[*returnSize] = k;
        (*returnSize)++;
    }
    return ans;
}