面试题32-2:把二叉树打印多行

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

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
class Solution {
public:
    /*
    思路:改编于面试题32,利用队列。
    先打印根节点,并把根节点左右子节点入队,打印队首元素并出队,将其左右孩子入队...循环如此一直到队列为空。
    此题要分层打印,则添加一个一维vector数组levelPrint存储每层的节点,再增设一个指针rightSon赋值为队列qTree尾元素,指向每层的最右节点
    当当前指针tNode与rightSon指向同一节点,表示该层遍历完毕,打印该层并重新设置levelPrint和rightSon节点
    */
        vector<vector<int> > Print(TreeNode* root) {
            //将vector作为返回值类型
            vector<vector<int> > printTree;
            //若为空树,直接返回空的数组
            if(root==nullptr)
                return printTree;
            //因为要分层打印,所以定义辅助指针指向每层最右边的节点,表示这是这一层的结尾
            TreeNode *rightSon=root;
            //储存每层的节点,最后加入到二维vector printTree中
            vector<int> levelPrint;
            //建立辅助队列
            queue<TreeNode*> qTree;
            TreeNode *tNode=root;
            //根节点入队
            qTree.push(tNode);
            //若队列不为空且树指针不指向空,执行循环:
            while(!qTree.empty()&&tNode!=nullptr)
            {
                //将队列首元素放入打印数组中
                levelPrint.push_back(tNode->val);

                //若队手元素左右子树不为空,压入队列
                if(tNode->left)
                    qTree.push(tNode->left);
                if(tNode->right)
                    qTree.push(tNode->right);
                //若此节点等于当前层最右节点,则将这一层结果输出并将最右节点重新赋值为队列尾结点
                if(rightSon==tNode)
                {
                    printTree.push_back(levelPrint);
                    levelPrint.clear();
                    rightSon=qTree.back();
                }
                //队首元素出队
                qTree.pop();
                //tNode赋值为新的队首元素
                tNode=qTree.front();
            }
            return printTree;
        }    
};
全部评论

相关推荐

点赞 评论 收藏
分享
我在朝九晚六双休的联想等你:如果我是你,身体素质好我会去参军,然后走士兵计划考研211只需要200多分。
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务