剑指offer--把二叉树打印成多行

2024年1月11日 284次阅读

题目描述

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

思路

乍一看和上一题很像，都是从左到右输出是不是就可以用层次遍历啦？
并不是！人家要求是每一层输出一行，一行一行的输出。而层次遍历是每遍历到一个就输出一个。
所以，还是得和上一题类似，用两个容器（队列）保存遍历的节点，交替使用。

C++代码

/* struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) { } }; */
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int>> results;
            if(pRoot == NULL){
                return results;
            }
            queue<TreeNode*> st1;
            queue<TreeNode*> st2;
            vector<int> row;
            row.push_back(pRoot->val);
            results.push_back(row);

            st1.push(pRoot);
            while(!st1.empty()||!st2.empty()){
                if(st1.empty()&&st2.empty()){
                    break;
                }
                vector<int> tmp;
                if(st2.empty()){
                    while(!st1.empty()){

                        if(st1.front()->left!=NULL){
                            tmp.push_back(st1.front()->left->val);
                            st2.push(st1.front()->left);
                        }
                        if(st1.front()->right!=NULL){
                            tmp.push_back(st1.front()->right->val);
                            st2.push(st1.front()->right);
                        }
                        st1.pop();
                    }
                }else{
                      while(!st2.empty()){
                         if(st2.front()->left!=NULL){
                            tmp.push_back(st2.front()->left->val);
                            st1.push(st2.front()->left);
                        }
                         if(st2.front()->right!=NULL){
                            tmp.push_back(st2.front()->right->val);
                            st1.push(st2.front()->right);
                        }
                        st2.pop();
                    }
                }             
                if(tmp.size()>0){
                    results.push_back(tmp);
                }

            }
            return results;  
        }
};
点赞