劍指offer--把二叉樹列印成多行(C++)

cherrydreamsover發表於2018-08-08
題目描述:從上到下按層列印二叉樹,同一層節點從左往右輸出,每一層輸出一行。
問題分析:這個問題就是二叉樹的層序遍歷,之前我的部落格中曾寫到過“鋸齒型層序列印二叉樹”,和本體思路基本一樣,並且本題相對那個題較簡單一些,我將那篇部落格的連線放在這裡:https://blog.csdn.net/cherrydreamsover/article/details/81051663
程式碼實現:
/*
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>> res;
            if(pRoot==NULL)
                return res;

            queue<TreeNode*> q;
            q.push(pRoot);

            while(!q.empty())
            {
                //用n來儲存每一次d佇列的元素個數,即就是二叉樹每一層節點的個數
                int n=q.size();
                vector<int> v;
                while(v.size() < n)
                {
                    TreeNode* pRoot=q.front();
                    q.pop();
                    v.push_back(pRoot->val);

                    if(pRoot->left)
                    {
                        q.push(pRoot->left);
                    }
                    if(pRoot->right)
                    {
                        q.push(pRoot->right);
                    }
                }
                res.push_back(v);
            }

            return res;
        }

};

相關文章