二叉树的按行打印及之字形打印

最新推荐文章于 2021-09-09 12:15:35 发布

咸鱼程序员

最新推荐文章于 2021-09-09 12:15:35 发布

阅读量879

点赞数 1

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/qq_37954088/article/details/81044192

数据结构同时被 2 个专栏收录

32 篇文章

订阅专栏

算法

12 篇文章

订阅专栏

本文介绍两种二叉树的遍历方法：层序遍历和之字形遍历。通过实例展示了如何使用队列实现这两种遍历方式，并提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 按行打印二叉树

分析：
按行打印二叉树，我们能想到的就是层序遍历二叉树。关键是要按行打印出来，我们可以定义两个变量，一个用来保存下一层元素的个数，一个用来表示当前行打印的元素个数。

class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {

        vector<vector<int> > vec;
        vector<int> vec2;

        if (pRoot == NULL)
            return vec;
        int nextLevel = 1;  //下一层元素的个数
        int tobePrint = 1;  //要打印的元素的个数

        queue<TreeNode*> q;
        q.push(pRoot);

        while (!q.empty())
        {
            tobePrint = nextLevel;
            nextLevel = 0;
            vec2.clear();

            while (tobePrint--)
            {
                TreeNode* pCur = q.front();
                vec2.push_back(pCur->val);
                if (pCur->left != NULL)
                {
                    q.push(pCur->left);
                    ++nextLevel;
                }
                if (pCur->right != NULL)
                {
                    q.push(pCur->right);
                    ++nextLevel;
                }
                q.pop();
            }
            vec.push_back(vec2);
        }
        return vec;
    }

};

2. 之字形打印二叉树

Given a binary tree, return the zigzag level order traversal of its nodes’ values. (ie, from left to right, then right to left for the next level and alternate between).
这里写图片描述

class Solution {
public:
    vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
        vector<vector<int> > result;

        if(root==NULL)
            return result;
        bool lefttoright=true;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
            int size=q.size();
            vector<int> vec(size);
            for(int i=0;i<size;i++)
            {
               TreeNode* pcur=q.front();
                q.pop();
                int index=(lefttoright)? i:(size-1-i);
                vec[index]=pcur->val;

                if(pcur->left)
                    q.push(pcur->left);
                if(pcur->right)
                    q.push(pcur->right);
            }
            lefttoright=!lefttoright;
            result.push_back(vec);
        }
         return result;   
    }
};