剑指Offer JZ38 二叉树的深度 C++实现

题目描述

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

解题思路

---

方法一：层次遍历

1、思路：用队列来对二叉树每一层进行遍历，这里用一个nullptr来记录每一层的结尾，当出队元素为nullptr且队列不为空时，说明已经遍历完上一层，且上一层的左右孩子结点也已经加入队列，此时层数加1，且将nullptr入队作为当前层的结尾。

2、代码：

/*
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;
        queue<TreeNode*> que;
        que.push(pRoot);
        que.push(nullptr);
        int level = 1;
        TreeNode *cur = nullptr;
        while (!que.empty()) {
            cur = que.front();
            que.pop();
            if (cur == nullptr) {
                if (que.empty()) break;
                level++;
                que.push(nullptr);
                continue;
            }
            if (cur->left) que.push(cur->left);
            if (cur->right) que.push(cur->right);
        }
        return level;
    }
};

3、复杂度：

时间复杂度：O(n)；

空间复杂度：O(n)。

---

方法二：递归

1、思路：对应二叉树先序遍历，遍历到叶子结点时，比较当前深度和历史最大深度，遍历完所有叶子结点后，返回最大深度。

2、代码：

class Solution {
public:
    void dfs(TreeNode *root, int &level, int &maxlevel) {
        if (!root) {
            if (level > maxlevel)
                maxlevel = level;
            return;
        }
        level++;
        dfs(root->left, level, maxlevel);
        dfs(root->right, level, maxlevel);
        level--;
    }
    int TreeDepth(TreeNode* pRoot)
    {
        if (!pRoot) return 0;
        int level = 0, maxlevel = INT_MIN;
        dfs(pRoot, level, maxlevel);
        return maxlevel;
    }
};

3、复杂度：

时间复杂度：O(n)；

空间复杂度：O(n)。

---

方法三：分治

1、思路：分别递归求左右子树的深度，回溯时返回左右子树中的最大深度，算上当前结点深度还要再加1。

2、代码：

class Solution {
public:
    int TreeDepth(TreeNode* pRoot)
    {
        if (!pRoot) return 0;
        int lval = TreeDepth(pRoot->left);
        int rval = TreeDepth(pRoot->right);
        return max(lval, rval) + 1;
    }
};

3、复杂度：

时间复杂度：O(n)；

空间复杂度：O(n)。