首页 > 其他 > 详细

Binary Tree Iterative Traversal

时间:2016-01-29 11:31:16      阅读:177      评论:0      收藏:0      [点我收藏+]

Preorder

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res;
        if (!root) return res;
        stack<TreeNode *> sta;
        sta.push(root);
        while (!sta.empty())
        {
            TreeNode* cur = sta.top();
            sta.pop();
            res.push_back(cur->val);
            if (cur->right) sta.push(cur->right);
            if (cur->left) sta.push(cur->left);
        }
        return res;
    }
};

 

postorder

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        if (!root) return res;
        stack<TreeNode *> sta;
        sta.push(root);
        TreeNode *last = NULL;
        while (!sta.empty())
        {
            TreeNode *cur = sta.top();
            TreeNode *left = cur->left, *right = cur->right;
            if (left && (!last || (last != left && last != right)))
                sta.push(left);
            else if (right && (!last || last != right))
                sta.push(right);
            else
            {
                res.push_back(cur->val);
                last = cur;
                sta.pop();
            }
        }
        return res;
    }
};

inorder

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> sta;
        while(1) {
            while(root) { sta.push(root); root = root->left; }
            if(sta.empty()) break;
            root = sta.top(); sta.pop();
            res.push_back(root->val);
            root = root->right;
        }
        return res;
    }
};

  

Binary Tree Iterative Traversal

原文:http://www.cnblogs.com/fenshen371/p/5168073.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!