NC14 按之字形顺序打印二叉树

题目描述:给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)

示例:

 

        该二叉树之字形层序遍历的结果是 [ [1], [3,2], [4,5] ]

解题思路:我们可以先了解一下二叉树的层序遍历,可以采用数据结构的堆来实现,可利用容器queue。首先将根结点加入队首,之后依次执行将队首的左右子节点加入队尾,并将队首pop弹出,直到队列为空。

        之字形的层序遍历与上述方法相似,此处的想法是可以借助两个栈结构来实现。两个栈分别储存奇数层和偶数层的结点。首先将根节点将入储存奇数层结点的栈,在每一层遍历结束的时候,如果是奇数层,则储存偶数层结点的栈应该为空;如果是偶数层,则储存奇数层结点的栈应该为空。如果偶数层节点的栈为空,则应取出储存奇数层结点的栈的首个元素,将其按右节点、左节点的顺序加入储存偶数层的栈,依次循环,直至存储奇数层的栈为空。如果储存奇数层结点的栈为空,则应取出偶数层结点的栈的首个元素,按期左节点、右节点的顺序加入储存奇数层结点的栈,直至储存偶数节点的栈为空。如果两个站都为空,则结束循环,即可完成之字形的层序遍历。

        代码实现如下:、

    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int>> ans;
        stack<TreeNode*>odd;
        stack<TreeNode*>even;
        even.push(pRoot);
        vector<int> an;
        if(pRoot==nullptr)
            return ans;
        while(odd.size()||even.size())
        {
            if(even.empty())
            {
                while(!odd.empty())
                {
                    TreeNode* pTreeNode = odd.top();
                    an.push_back(pTreeNode->val);
                    odd.pop();
                    if(pTreeNode->right)
                        even.push(pTreeNode->right);
                    if(pTreeNode->left)
                        even.push(pTreeNode->left);
                }
                ans.push_back(an);
                an.clear();
            }
            else if(odd.empty())
            {
                while(!even.empty())
                {
                    TreeNode* pTreeNode = even.top();
                    an.push_back(pTreeNode->val);
                    even.pop();
                    if(pTreeNode->left)
                        odd.push(pTreeNode->left);
                    if(pTreeNode->right)
                        odd.push(pTreeNode->right);
                }
                ans.push_back(an);
                an.clear();
            }
        }
        return ans;
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值