剑指offer-二叉树的深度

题目：

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。

链接：

http://www.nowcoder.com/practice/435fb86331474282a3499955f0a41e8b?tpId=13&tqId=11191&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

代码：

心血来潮用栈实现了一下树的后序递归遍历

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    int TreeDepth(TreeNode* pRoot)
    {
        if(pRoot == NULL){
            return 0;
        }
        stack<TreeNode*> st;
        stack<bool> fst;
        TreeNode *top = pRoot;
        bool ftop;
        int maxDepth = 0;
        while(top != NULL || !st.empty()){
            while(top != NULL){
                st.push(top);
                fst.push(true);
                top = top->left;
            }

            top = st.top();
            ftop = fst.top();
            
            st.pop();
            fst.pop();
            if(ftop){
                st.push(top);
                fst.push(false);
                
                if(top->right == NULL && st.size() > maxDepth){
                    maxDepth = st.size();
                }
                top = top->right;
            }else{
                top = NULL;
            }
            
        }
        
        return maxDepth;
    }
};