把二叉树打印成多行

发表于 | 分类于 | 评论数： | 阅读次数： | 阅读次数：

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
vector<vector<int> > Print(TreeNode* pRoot) {
    
    vector<int> level;
    if (pRoot == nullptr) {
        return res;
    }
    queue<TreeNode*> que;
    que.push(pRoot);
    while (!que.empty()) {
        vector<vector<int>> res;
        int n = que.size();
        for (int i = 0; i < n; i++) {
            TreeNode* node = que.front();
            que.pop();
            level.push_back(node->val);
            if (node->left) {
                que.push(node->left);
            }
            if (node->right) {
                que.push(node->right);
            }
        }
             
            res.push_back(level);
        
        
        }
     
    return res;           
 }

这里只用一个队列就可以了， 注意进入while循环时， 先计算队列的大小， 然后用for循环处理完这层的所有元素后再转到下一层的元素。

清空vector中的所有元素

1
2


0%