leetcode第一刷_Binary Tree Postorder Traversal

来源:互联网 发布:ubuntu完全卸载wine 编辑:程序博客网 时间:2024/04/30 12:28

树的结构决定了有关他的问题使用递归是最方便的,二叉树来说,父亲,左孩子右孩子,多么清晰明了。但是这也造成了我的思维定式,如果不用递归解决树的问题,我还真不是很会写,即使是最简单的这种遍历问题。

递归转非递归一般得借助栈。后序遍历是先访问左右孩子,再访问根节点,我们在访问左右孩子之前一定是先经过父亲节点了,那么怎么办呢,先把父亲节点放到栈里,每次出栈的时候再访问它。树最左下的叶子节点一定是先被访问的,应该先沿着左孩子这条线往下找到这个节点,然后看看它有没有右孩子,如果有右孩子,说明它还不够左下,从右孩子的左孩子这条线继续找。获得栈顶元素时,不能马上出栈,要看看它有没有右孩子,如果有右孩子,且没有被访问过,那么应当继续从它右孩子那里走,如果访问过了,说明当前节点的左右孩子都访问完毕了,可以将其出栈。这个访没访问过,并不需要单独开一个数组来记录状态,只需要记录上个访问的节点,因为访问过右孩子又回到父亲的这种情况,上一个访问的一定是最近的这个右孩子。一个指针就可以了。

/** * 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> v;        stack<TreeNode*> s;        TreeNode *pre = NULL;        TreeNode *pNode = root;        while(pNode || !s.empty()){            while(pNode){                s.push(pNode);                pNode = pNode->left;            }            if(!s.empty()){                TreeNode *t = s.top();                if(t->right && t->right != pre){                    pNode = t->right;                }else{                    v.push_back(t->val);                    pre = t;                    s.pop();                }            }        }        return v;    }};


0 0
原创粉丝点击