剑指offer.从上到下打印二叉树

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

思路:

乍一看这个题其实就是考的二叉树的遍历,但是属于特殊的层序遍历。对于没有接触过层序遍历的人来说最简单的想法肯定是设计两个函数,一个打印函数一个是遍历函数,先打印根节点,然后打印根节点的左子节点,再打印根结点的右子节点,然后root=root->left;但这个遍历有一个问题就是怎么在root=root->left后再root=root->right,再设置一个变量来储存根节点?显然不显示,对于上层的根节点还好解释,往下层走情况就会变得很复杂了。显然这个思路是行不通的。

所以我们需要用别的媒介来改变这一遍历方式。比如创建一个容器来存储这些节点值。接下来我们大概梳理一下这个思路过程

1.按照题目的意思我们还是按层打印,先打印根节点,为了接下来能够打印根节点的两个子节点,我们遍历到根节点时,我们就把那两个子节点存入,按照从左到右的存储顺序,然后再从容器里取出来,也是从左到右的取出来。

2.所以我们这个容器很显然易见,先进先出,就是队列。打印出根节点的左子节点后就再把左子节点的两个子节点再存入队列中,后面的思路就是基本一致。

代码实现:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */


/**
 * Note: The returned array must be malloced, assume caller calls free().
 */

struct Quene {
    struct TreeNode stackin[1000];
    struct TreeNode stackout[1000];
    struct TreeNode* top_in;
    struct TreeNode* top_out;
};

struct Quene* CreatQuene()
{
    struct Quene* ret = (struct Quene*)malloc(sizeof(struct Quene));
    ret->top_in = ret->stackin;
    ret->top_out = ret->stackout;
    return ret;
}

void Push(struct TreeNode* root, struct Quene* quene)
{
    *(quene->top_in) = *root;
    (quene->top_in)++;
}

struct TreeNode Pop(struct Quene* quene)
{
    if (quene->top_out == quene->stackout) {
        if (quene->top_in == quene->stackin) {
        } else {
            while (quene->top_in != quene->stackin) {
                (quene->top_in)--;
                *(quene->top_out) = *(quene->top_in);
                (quene->top_out)++;
            }
        }
    }
    (quene->top_out)--;
    return *(quene->top_out);
}

bool IsEmpty(struct Quene* quene)
{
    if ((quene->stackin == quene->top_in) && (quene->stackout == quene->top_out)) {
        return true;
    }
    return false;
}

int* levelOrder(struct TreeNode* root, int* returnSize)
{
    if (root == NULL) {
        *returnSize = 0;
        return NULL;
    }
    struct Quene* quene = CreatQuene();
    Push(root, quene);
    int buf[10000];
    int q = 0;
    
    while (!IsEmpty(quene)) {
        struct TreeNode temp = Pop(quene);
        buf[q++] = temp.val;
        if (temp.left != NULL) {
            Push(temp.left, quene);
        }
        if (temp.right != NULL) {
            Push(temp.right, quene);
        }
    }
    *returnSize = q;
    int* ret = (int*)malloc(q * sizeof(int));
    memcpy(ret, buf, q * sizeof(int));
    return ret;
}

模拟队列:

int* levelOrder(struct TreeNode* root, int* returnSize){
    if(!root){
        *returnSize=0;
        return NULL;
    }
    *returnSize=0;
    int* res=(int*)malloc(sizeof(int)*10000);
    struct TreeNode** queue=(struct TreeNode**)malloc(sizeof(struct TreeNode*)*10000);
    int head=0,rear=0;
    queue[rear++]=root;
    while(head<rear){
        struct TreeNode* tmp=queue[head++];
        res[(*returnSize)++]=tmp->val;
        if(tmp->left){
            queue[rear++]=tmp->left;
        }
        if(tmp->right){
            queue[rear++]=tmp->right;
        }
    }
    return res;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值