题解 | #牛牛的N叉树#

牛牛的N叉树

https://www.nowcoder.com/practice/123742d6dac042d08521f3d0edf68d22

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型
     */
    int maxDepth(TreeNode* root) {
        // write code here
        // 采用BFS
        queue<TreeNode*> q1;
        queue<TreeNode*> q2;
        int deep = 0;
        if(root) {
            q1.push(root);
        }
        while(!q1.empty()){
            deep++;
            // all son
            int size = q1.size();
            for(int i = 0; i < size; ++i){  // push all next layer node bro
                TreeNode* temp = q1.front()->left;  // 子女
                while(temp){
                    q2.push(temp);
                    temp = temp->right;  // 兄弟
                }
                q1.pop();
            }
            q1.swap(q2);
        }
        return deep;
    }
};

全部评论

相关推荐

09-27 14:42
已编辑
浙江大学 Java
未来未临:把浙大放大加粗就行
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务