二叉樹的深度

題目

輸入一棵二叉樹，求該樹的深度。從根結點到葉結點依次經過的結點（含根、葉結點）形成樹的一條路徑，最長路徑的長度爲樹的深度。

思路

前序遍歷二叉樹，記錄一下深度。

參考代碼

/*
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) return 0;
        int sum = 0;
        getDepth(pRoot, 0, sum);
        return sum;
    }
private:
    void getDepth(TreeNode* root, int cnt, int& depth) {
        if (!root) {
            if (cnt > depth) depth = cnt;
            return;
        }
        cnt++;
        getDepth(root->left, cnt, depth);
        getDepth(root->right, cnt, depth);
    }
};