二叉树 层次交替遍历二叉树

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,

第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

思路:层次遍历采用队列模拟

开始 根节点入队列

开始一个循环,每次出队列一个元素,访问该值

若该节点 左右子节点不为NULL 放入队列尾部

循环直到 队列为空

——如何交替一行一行输出 一开始队列的长度为size表示上一层节点数目 每次控制访问size个,

另外增加一个flag控制左右顺序

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};

 核心函数

vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int>> res;
        if(pRoot==NULL)
            return res;
        vector<TreeNode*> queue;//模拟队列
        //首先根节点入队列
        queue.push_back(pRoot);
        bool flag=false;//从左向右
        while(queue.empty()!=1)
        {
            int size=queue.size();//该层节点数目
            vector<int> temp;
            int i=0;
            while(i<size)
            {
                TreeNode* p=queue[0];//取队列头
                temp.push_back(p->val);
                if(p->left!=NULL)
                    queue.push_back(p->left);
                if(p->right!=NULL)
                    queue.push_back(p->right);
                queue.erase(queue.begin());//出队列
                i++;
                
            }
            if(flag)
            {
                vector<int> one_res;
                for(int i=temp.size()-1;i>=0;i--)
                {
                     one_res.push_back(temp[i]);
                }
                res.push_back(one_res);
                flag=false;//下一次从右向左
           
            }
            else
            {
                res.push_back(temp);
                flag=true;
            }
        
        }
        return res;      
    }

猜你喜欢

转载自blog.csdn.net/qq_33369979/article/details/87870629