码迷,mamicode.com
首页 > 其他好文 > 详细

剑指offer61 按之字形顺序打印二叉树

时间:2017-07-24 22:30:41      阅读:124      评论:0      收藏:0      [点我收藏+]

标签:off   bsp   root   解法   result   clear   vector   pre   二叉树   

另一种自己写的解法

class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int> > result;
        vector<int> res;
        if(pRoot == NULL)
            return result;
        stack<TreeNode* > sta1;
        stack<TreeNode* > sta2;
        int next = 2;
        sta1.push(pRoot);
        while(sta1.size() != 0 || sta2.size() != 0){
            if(next % 2 == 0){
                if(sta1.empty()){
                    next++;
                    result.push_back(res);
                    res.clear();
                }
                else{
                    TreeNode* Node = sta1.top();
                    res.push_back(Node->val);
                    if(Node->left != NULL)
                        sta2.push(Node->left);
                    if(Node->right != NULL)
                        sta2.push(Node->right);
                    sta1.pop();
                }
            }
            else{
                if(sta2.empty()){
                    next++;
                    result.push_back(res);
                    res.clear();
                }
                else{
                    TreeNode* Node = sta2.top();
                    res.push_back(Node->val);
                    if(Node->right != NULL)
                        sta1.push(Node->right);
                    if(Node->left != NULL)
                        sta1.push(Node->left);
                    sta2.pop();
                }
            }
        }
        result.push_back(res);
        return result;
    }
};

 

剑指offer61 按之字形顺序打印二叉树

标签:off   bsp   root   解法   result   clear   vector   pre   二叉树   

原文地址:http://www.cnblogs.com/ymjyqsx/p/7231222.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!