美文网首页
剑指offer 32- 分行从上往下打印二叉树

剑指offer 32- 分行从上往下打印二叉树

作者: 顾子豪 | 来源:发表于2021-05-21 11:19 被阅读0次

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

输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
    8
   / \
  12  2
     /
    6
   /
  4

输出:[[8], [12, 2], [6], [4]]

分析:
每一层结束的时候,往queue里塞一个NULL做标记。

在queue里读取一个数出来之后,先看看是不是行标识符NULL
(因为是BFS,当前行读完,下一行有哪些要读的也都放在queue里了,可以在queue结尾给加一个新的NULL), 是的话再看看是不是整个树读完了(即queue里没有点了)。

时间复杂度分析:每个点遍历一次,所以时间复杂度是 O(n)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> printFromTopToBottom(TreeNode* root) {
        vector<vector<int>> res;
        queue<TreeNode*> q;
        if (!root) return res;
        q.push(root);
        q.push(nullptr);//存放NULL作为当前行的行末标记
        vector<int> cur; // 用于存放当前行的元素
        while(q.size()) {
            auto x = q.front();
            q.pop();
            if (x) {
                cur.push_back(x->val);
                if(x->left) q.push(x->left);
                if(x->right) q.push(x->right);
            } else {//访问到行末标记null
                if(q.size()) q.push(nullptr); //增加行末标记
                res.push_back(cur); //当前行存放结果数组中
                cur.clear(); //把当前行清空
            }
            
        }
        return res;
    }
};

相关文章

网友评论

      本文标题:剑指offer 32- 分行从上往下打印二叉树

      本文链接:https://www.haomeiwen.com/subject/gwzljltx.html