二叉树的层次遍历 & 深度

题目:二叉树的层次遍历

层序遍历就是按二叉树从上到下,从左到右依次打印每个节点中存储的数据。

思路标签:

  • 数据结构:队列

解答:

  • 二叉树的层次遍历满足队列先入先出的规则;
  • 每次访问队列头部节点,将其出队列,并将其左右子节点分别入队列。
struct BinaryTree {
    int val;
    BinaryTree *left;
    BinaryTree *right;
    BinaryTree(int value)
        : val(value), left(nullptr), right(nullptr) { }
};

void levelOrder(BinaryTree *root) {
    if (root == nullptr)
        return;

    queue<BinaryTree*> que;
    que.push(root);
    while (!que.empty()) {
        BinaryTree *front = que.front();
        cout << front->val << " ";
        que.pop();
        if (front->left != nullptr)
            que.push(front->left);
        if (front->right != nullptr)
            que.push(front->right);
    }
} 

题目:二叉树的深度(层次遍历)

利用层次遍历可以得到二叉树的深度

思路标签:

  • 数据结构:队列

解答:

  • 二叉树的层次遍历满足队列先入先出的规则;
  • 每次访问队列头部节点,将其出队列,并将其左右子节点分别入队列。
struct BinaryTree {
    int val;
    BinaryTree *left;
    BinaryTree *right;
    BinaryTree(int value)
        : val(value), left(nullptr), right(nullptr) { }
};

int treeDepth(BinaryTree* root) {
    if (root == nullptr) return 0;

    int depth = 0;
    queue<BinaryTree*> que;
    que.push(root);
    while (!que.empty()) {
        depth++;
        int count = 0;
        int size = que.size();

        while (count < size) { //弹出当前层的所有节点
            BinaryTree *front = que.front();
            que.pop();
            count++;
            if (front->left)
                que.push(front->left);
            if (front->right)
                que.push(front->right);
        }
    }

    return depth;
}

猜你喜欢

转载自blog.csdn.net/koala_tree/article/details/79991074