剑指offer-38 -- 二叉树的深度 - C++

题目描述

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

解题：层次遍历法

求最大深度，可用队列。因为要满足先进先出的特性。

1. 初始化：一个队列queue<TreeNode*> q， 将root节点入队列q
2. 如果队列不空，做如下操作：
3. 弹出队列头，保存为node，将node的左右非空孩子加入队列
4. 做2,3步骤，知道队列为空

如果不需要确定当前遍历到了哪一层，模板如下：

void bfs() {
    vis[] = 0;
    queue<int> pq(start_val);

    while (!pq.empty()) {
        int cur = pq.front(); pq.pop();
        for (遍历cur所有的相邻节点nex) {
            if (nex节点有效 && vis[nex]==0){
                vis[nex] = 1;
                pq.push(nex)
            }
        }
    }
}

如果需要确定遍历到哪一层，模板如下：

void bfs() {
    int level = 0;
    vis[] = 0; // or set
    queue<int> pq(original_val);
    while (!pq.empty()) {
        int sz = pq.size();

        while (sz--) {
                int cur = pq.front(); pq.pop();
            for (遍历cur所有的相邻节点nex) {
                if (nex节点有效 && vis[nex] == 0) {
                    vis[nex] = 1;
                    pq.push(nex)
                }
            } // end for
        } // end inner while
        level++;

    } // end outer while
}

所以本题直接套模板即可：

代码如下：

class Solution {
public:
    int TreeDepth(TreeNode* pRoot)
    {
        int res = 0;
        queue<TreeNode*> d;
        if(!pRoot)
            return 0;
        d.push(pRoot);
        while(!d.empty()){
            int sz = d.size();
            while(sz){
                TreeNode* Node = d.front();
                d.pop();
                if(Node->left)
                    d.push(Node->left);
                if(Node->right)
                    d.push(Node->right);
                sz -= 1;
            }
            res += 1;
        }
        return res;
    }
};

时间复杂度：O(n),二叉树的每个节点遍历一次
空间复杂度：O(n),最差情况下，树平衡时，队列最多存储n/2个节点。

类似题目

把二叉树打印成多行：https://blog.csdn.net/laobai1015/article/details/112376349

从上往下打印二叉树：https://blog.csdn.net/laobai1015/article/details/108721181

按之字形顺序打印二叉树：https://blog.csdn.net/laobai1015/article/details/112376105