leetcode Binary Tree Postorder Traversal

265 篇文章 1 订阅
231 篇文章 0 订阅

Given a binary tree, return the postorder traversal of its nodes' values.

For example:
Given binary tree {1,#,2,3},

   1
    \
     2
    /
   3

return [3,2,1].

Note: Recursive solution is trivial, could you do it iteratively?

Solution1 : Simulate the recursive program:

Note that: 1. sta.pop();

                2. 

      if (!sta.empty()) {
not 

      while (!sta.empty()) {
The right code is :

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
 public:
  vector<int> postorderTraversal(TreeNode *root) {
    stack<pair<TreeNode*, int>> sta;
    vector<int> res;
    while (root != NULL || !sta.empty()) {
      while (root != NULL) {
        sta.push(make_pair(root, 0)); //0 stands for right child and itself;
        root = root->left;
      }    
      if (!sta.empty()) {
        root = sta.top().first;
        int status = sta.top().second;
        sta.pop();
        if (status == 0) {
          sta.push(make_pair(root, 1));
          root = root->right;
        }
        else if (status == 1){
          res.push_back(root->val);
          root = NULL;
        }
      }
    }
    return res;
  }
};

Solution 2: Using two stacks:

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
 public:
  vector<int> postorderTraversal(TreeNode *root) {
    stack<TreeNode*> s,output;
    vector<int> res;
    TreeNode* cur = NULL;
    if (root != NULL)
      s.push(root);
    while (!s.empty()) {
      cur = s.top();
      s.pop();
      output.push(cur);
      if (cur->left)
        s.push(cur->left);
      if (cur->right)
        s.push(cur->right);
    }
    while (!output.empty()) {
      res.push_back(output.top()->val);
      output.pop();
    }
    return res;
  }
};

Solution 3: Record the previous node, when cur->left == prev, push the right child to stack for the next-turn visiting:

A wrong code:

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
 public:
  vector<int> postorderTraversal(TreeNode *root) {
    vector<int> res;
    TreeNode *prev = NULL, *cur = NULL;
    stack<TreeNode*> s;
    if (root != NULL)
      s.push(root);
    while (!s.empty()) {
      cur = s.top();
      if (prev == NULL || prev->left == cur || prev->right == cur) 
        if (cur->left)
          s.push(cur->left);
        else if (cur->right)
          s.push(cur->right);
      else if (cur->left == prev && cur->right)
        s.push(cur->right);
      else {
        res.push_back(cur->val);
        s.pop();
      }
      prev = cur;
    }
    return res;
  }
};

The right code is :

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
 public:
  vector<int> postorderTraversal(TreeNode *root) {
    vector<int> res;
    TreeNode *prev = NULL, *cur = NULL;
    stack<TreeNode*> s;
    if (root != NULL)
      s.push(root);
    while (!s.empty()) {
      cur = s.top();
      if (prev == NULL || prev->left == cur || prev->right == cur) {
        if (cur->left)
          s.push(cur->left);
        else if (cur->right)
          s.push(cur->right);
      }
      else if (cur->left == prev && cur->right)
        s.push(cur->right);
      else {
        res.push_back(cur->val);
        s.pop();
      }
      prev = cur;
    }
    return res;
  }
};



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值