微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

二叉树的深度

递归(深度遍历dfs)
时间复杂度:O(n),遍历二叉树每个结点
空间复杂度:O(n),递归栈深度就是二叉树的高度,其中最坏情况是二叉树退化为链表,深度最大为n

/*
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==nullptr) return 0;
        int lDepth = TreeDepth(pRoot->left);
        int rightDepth = TreeDepth(pRoot->right);
        return max(lDepth, rightDepth) + 1;
    
    }
};

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

层次遍历(dfs)
时间复杂度:O(n),遍历二叉树每个结点
空间复杂度:O(n),辅助队列的最大长度不会超过n

class Solution {
public:
    int TreeDepth(TreeNode* pRoot) {
        if(pRoot==nullptr) return 0;
        queue<TreeNode*> q;
        q.push(pRoot);
        int level = 0;
        while(!q.empty()){
            int size = q.size();
            while(size--){
                auto node = q.front();
                q.pop();
                if(node->left) q.push(node->left);
                if(node->right) q.push(node->right);
            }
            leveL++;
        }
         return level;
    }
};
class Solution {
public:
    int TreeDepth(TreeNode* pRoot) {
        if(pRoot==nullptr) return 0; //空结点没有深度
        queue<TreeNode*> q; // 生成队列
        q.push(pRoot); //根入队
        int level = 0; //记录深度 
        while(!q.empty()){ // bfs
            int size = q.size(); //记录当前层有多少结点
            for (int i=0; i<size; i++) {
                auto node = q.front(); // TreeNode* node = q.front(); 
                q.pop();
                if(node->left) q.push(node->left);
                if(node->right) q.push(node->right);
            }
            leveL++; // 每访问完一层结点,深度加1
        }
         return level;
    }
};

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐