leetcode102 剑指offer 面试题32 - II. 从上到下打印二叉树 II

题干:
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
   
返回其层次遍历结果:
[
  [3],
  [9,20],
  [15,7]
]

提示:
节点总数 <= 1000

解题思路:
核心步骤和之前的面试题31差不多,不同的地方在于,打印方式是一个二维数组,且每行的长度不同,用c语言比较难以处理,要不断的赋值和更新行长度。

代码实现:

#define INT_MAX 1010
int** levelOrder(struct TreeNode* root, int* returnSize, int** returnColumnSizes){
    if (root == NULL){          //判断树是否存在
        * returnSize = 0;
        return NULL;
    }
    //定义一个指针队列存储所有节点的地址,并分配每一列的内存,定义一个二维数组做输出
    struct TreeNode** Queue = (struct TreeNode**)calloc(INT_MAX, sizeof(struct TreeNode*));
    * returnColumnSizes = (int*)calloc(INT_MAX, sizeof(int));
    int** nums = (int**)calloc(INT_MAX, sizeof(int*));
    //初始化各项参数,包括首尾指针,行数
    int head = 0;
    int tail = 0;
    * returnSize = 0;
    
    Queue[tail++] = root;                   //根节点入队
    int rowSize = tail - head;
    while (rowSize){
        nums[* returnSize] = (int*)calloc(rowSize, sizeof(int));    //分配每一行的存储空间
        (* returnColumnSizes)[* returnSize] = rowSize;     
        for (int i = 0; i < rowSize; i++){
            if (Queue[head] -> left != NULL){
                Queue[tail++] = Queue[head] -> left;
            }
            if (Queue[head] -> right != NULL){
                Queue[tail++] = Queue[head] -> right;
            }
            nums[* returnSize][i] = Queue[head] -> val;
            head++;
        }
        (* returnSize)++;                                       //行数加一
        rowSize = tail - head;                                  //计算每一层的节点数
    }
    return nums;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值