二叉树的层序遍历_二叉树层序遍历

二叉树的层序遍历比较基础，想法很简单，就是用一个queue就可以实现，一次把弹出节点的左右子节点放入queue中，然后读出就行。

1.简单的二叉树层序遍历

class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        vector<int> res;
        if(root==NULL)
            return res;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty())
        {
            TreeNode* t=que.front();
            que.pop();
            res.push_back(t->val);
            if(r->left!=nullptr) que.push(r->left);
            if(t->right!=nullptr) que.push(r->right);
        }
        return res;
    }
};

2.二叉树层序遍历并逐层打印（每一层输出一行）

注意，这个是要每层输出一行，难点就是控制行的打印。

class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int> > res;
            if(pRoot==NULL)
                return res;
            queue<TreeNode*> que;
            que.push(pRoot);//先将头结点放入

            while(!que.empty())
            {
                int s=0;
                int node_size=que.size();//这一步比着上一步是最重要的区别，目的是
                //先读出queue里面还有多少个节点，这个节点数就是每层的数，下面每层打印的时候
                //会把打印完的一层逐个扔掉。

                vector<int> r;
                while(s++<node_size)//s从0开始直到将queue中的数遍历完。
                {
                    TreeNode* t=que.front();
                    que.pop();
                    r.push_back(t->val);
                    if(t->left) que.push(t->left);
                    if(t->right) que.push(t->right);
                }
                res.push_back(r);
            }
            return res;
        
        }
};

3.之字型打印二叉树

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推

解：和前面思路一样，不同的是用两个栈就行了，在这里其实不用计数，直接是a弹出完结束，b弹出完结束。

class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int> > res;
        if(pRoot==NULL)
            return res;
        stack<TreeNode*> a;
        stack<TreeNode*> b;
        a.push(pRoot);
        while(!a.empty()||!b.empty())
        {
            int count1=0;
            int nodeInStack1=a.size();
            if(nodeInStack1==0)
                break;
            vector<int> r1;
            while(count1++<nodeInStack1)//不用计数，直接while(!a.empty())就可以了
            {
                
                TreeNode* t=a.top();
                a.pop();
                r1.push_back(t->val);
                if(t->left) b.push(t->left);
                if(t->right) b.push(t->right);
            }
            res.push_back(r1);
            
            int count2=0;
            int nodeInStack2=b.size();
            if(nodeInStack2==0)
                break;
            vector<int> r2;
            while(count2++<nodeInStack2)
            {
                
                TreeNode* t=b.top();
                b.pop();
                r2.push_back(t->val);
                if(t->right) a.push(t->right);
                if(t->left) a.push(t->left);
            }
            res.push_back(r2);
        }
        return res;
        
        
        
    }
    
};