剑指Offer-从上到下打印二叉树

1. 题目 1

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

1.1 示例

输入:

     8
    / \
   6   10
  / \  / \
 5   7 9  11

输出:

8 6 10 5 7 9 11

1.2 解题思路

这个其实就是层序遍历。每次打印一个结点的时候,若该结点有子结点,则让该结点的子结点放到一个队列的末尾。接下来到队列的首部取出最早进入队列的结点,不断重复前面的打印,直至队列中所有的结点都被打印出来。

1.3 代码实现

#include <iostream>
#include <vector>
#include <queue>

struct BiTNode {
    int val;
    BiTNode* left;
    BiTNode* right;
    BiTNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

class Solution {
public:
    std::vector<int> PrintFromTopToBottom(BiTNode* root) {
        // 存储打印结果
        std::vector<int> result;
        
        // 边界条件
        if (root == nullptr) {
            return result;
        }
        
        // 辅助容器:队列
        std::queue<BiTNode*> qeueTreeNode;
        
        // 根结点入队
        qeueTreeNode.push(root);
        
        // 遍历队列
        while (!qeueTreeNode.empty()) {
            // 辅助指针:指向队头
            BiTNode* node = qeueTreeNode.front();

            // 打印结果存储到result中
            result.push_back(node->val);
            
            // 根结点的子结点入队
            if (node->left != nullptr) {
                qeueTreeNode.push(node->left);
            }
            if (node->right != nullptr) {
                qeueTreeNode.push(node->right);
            }

            // 根结点出队
            qeueTreeNode.pop();
        }
        
        return result;
    }
};

// 销毁二叉树
void DestroyBiTree(BiTNode* root)
{
    if (root == nullptr) {
        return;
    }

    DestroyBiTree(root->left);
    DestroyBiTree(root->right);

    // 从最左依次 delete 至根结点
    delete root;
    root = nullptr;        
}

int main(void)
{
    Solution sol;
    std::vector<int> res;
    
    // 构造一个二叉树
    BiTNode* root = new BiTNode(8);
    root->left = new BiTNode(6);
    root->right = new BiTNode(10);
    root->left->left = new BiTNode(5);
    root->left->right = new BiTNode(7);
    root->right->left = new BiTNode(9);
    root->right->right = new BiTNode(11);
    
    res = sol.PrintFromTopToBottom(root);
    
    // 遍历vector,C++ 11 标准支持
    for (int i:res) {
         std::cout << i;
         
        if (i != res[res.size() - 1]) {
            std::cout << ' ';
        }
    }
    
    // 遍历vector,打印结果
    // for (unsigned int i = 0; i < res.size(); ++i) {
    //     std::cout << res[i];
    //     if (i != res.size() - 1) {
    //         std::cout << ' ';
    //     }
    // }
    std::cout << std::endl;

    DestroyBiTree(root);
    
    return 0;
}

2. 题目 2

分行从上到下打印二叉树。从上到下按层打印二叉树,同一层的结点按从左到右的顺序打印,每一层打印到一行。

2.1 示例

输入:

     8
    / \
   6   10
  / \  / \
 5   7 9  11

输出:

8
6 10
5 7 9 11

2.2 解题思路

为了把二叉树的每一行单独打印到一行里,我们需要两个变量:一个变量表示当前层中还未打印的结点数;另一个变量表示下一层结点的数目。

2.3 代码实现

#include <iostream>
#include <queue>

struct BiTNode {
    int val;
    BiTNode* left;
    BiTNode* right;
    BiTNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

class Solution {
public:
    int PrintTreesInLines(BiTNode* root) {
 
        // 边界条件
        if (root == nullptr) {
            return -1;
        }

        std::queue<BiTNode*> qeueTreeNode;

        qeueTreeNode.push(root);

        // 下一层的结点数
        int nextLevel = 0;

        // 当前层中未打印的结点数
        int toBePrinted = 1;

        while (!qeueTreeNode.empty()) {
            BiTNode* node = qeueTreeNode.front();
            
            // 打印结果
            std::cout << node->val;

            // 打印结果之间用空格隔开
            if (toBePrinted != 1) {
                std::cout << ' ';
            }

            if (node->left != nullptr) {
                qeueTreeNode.push(node->left);
                ++nextLevel;
            }
            if (node->right != nullptr) {
                qeueTreeNode.push(node->right);
                ++nextLevel;
            }

            qeueTreeNode.pop();
            --toBePrinted;
            
            // 当前层的所有结点已打印完毕,可以继续打印下一层
            if (toBePrinted == 0) {
                std::cout << std::endl;
                toBePrinted = nextLevel;
                nextLevel = 0;
            }
        }
        
        return 0;
    }
};

// 销毁二叉树
void DestroyBiTree(BiTNode* root)
{
    if (root == nullptr) {
        return;
    }

    DestroyBiTree(root->left);
    DestroyBiTree(root->right);

    // 从最左依次 delete 至根结点
    delete root;
    root = nullptr;        
}

int main(void)
{
    Solution sol;
    
    // 构造一个二叉树
    BiTNode* root = new BiTNode(8);
    root->left = new BiTNode(6);
    root->right = new BiTNode(10);
    root->left->left = new BiTNode(5);
    root->left->right = new BiTNode(7);
    root->right->left = new BiTNode(9);
    root->right->right = new BiTNode(11);
    
    sol.PrintTreesInLines(root);

    std::cout << std::endl;
    
    DestroyBiTree(root);
    
    return 0;
}

3. 题目 3

之字形打印二叉树。请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二行按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其它行以此类推。

3.1 示例

输入:

          1
       /      \
    2           3
   / \        /   \
  4    5     6     7
/  \  /  \   / \  / \
8  9  10 11 12 13 14 15

输出:

1
3 2
4 5 6 7
15 14 13 12 11 10 9 8

3.2 解题思路

按之字形顺序打印二叉树需要两个栈。在打印某一层的结点时,把下一层的子结点保存到相应的栈里。若当前打印的是奇数层(第 1 、3 层等),则先保存左子结点再保存右子结点到第一个栈里;若打印的是偶数层(第 2 、4层等),则先保存右子结点再保存左子结点到第二个栈里。

3.3 代码实现

#include <iostream>
#include <stack>

struct BiTNode {
    int val;
    BiTNode* left;
    BiTNode* right;
    BiTNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

class Solution {
public:
    int PrintTreesInZigzag(BiTNode* root) {
 
        // 边界条件
        if (root == nullptr) {
            return -1;
        }

        std::stack<BiTNode*> stackLevels[2];

        int current = 0;
        int next = 1;

        stackLevels[current].push(root);

        while (!stackLevels[0].empty() || !stackLevels[1].empty()) {
            BiTNode* node = stackLevels[current].top();
            stackLevels[current].pop();

            std::cout << node->val;

            if (!stackLevels[current].empty()) {
                std::cout << ' ';
            }

            if (current == 0) {
                if (node->left != nullptr) {
                    stackLevels[next].push(node->left);
                }
                if (node->right != nullptr) {
                    stackLevels[next].push(node->right);
                }
            } else {
                if (node->right != nullptr) {
                    stackLevels[next].push(node->right);
                }
                if (node->left != nullptr) {
                    stackLevels[next].push(node->left);
                }
            }

            if (stackLevels[current].empty()) {
                std::cout << std::endl;
                current = 1 - current;
                next = 1 - next;
            }
        }
        
        return 0;
    }
};

// 销毁二叉树
void DestroyBiTree(BiTNode* root)
{
    if (root == nullptr) {
        return;
    }

    DestroyBiTree(root->left);
    DestroyBiTree(root->right);

    // 从最左依次 delete 至根结点
    delete root;
    root = nullptr;        
}

int main(void)
{
    Solution sol;
    
    // 构造一个二叉树
    BiTNode* root = new BiTNode(1);
    root->left = new BiTNode(2);
    root->right = new BiTNode(3);
    root->left->left = new BiTNode(4);
    root->left->right = new BiTNode(5);
    root->right->left = new BiTNode(6);
    root->right->right = new BiTNode(7);
    root->left->left->left = new BiTNode(8);
    root->left->left->right = new BiTNode(9);
    root->left->right->left = new BiTNode(10);
    root->left->right->right = new BiTNode(11);
    root->right->left->left = new BiTNode(12);
    root->right->left->right = new BiTNode(13);
    root->right->right->left = new BiTNode(14);
    root->right->right->right = new BiTNode(15);
    
    sol.PrintTreesInZigzag(root);

    std::cout << std::endl;
    
    DestroyBiTree(root);
    
    return 0;
}

个人主页:

www.codeapes.cn

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 200,176评论 5 469
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 84,190评论 2 377
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 147,232评论 0 332
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,953评论 1 272
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,879评论 5 360
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,177评论 1 277
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,626评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,295评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,436评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,365评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,414评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,096评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,685评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,771评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,987评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,438评论 2 346
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,032评论 2 341