【Tree】BFS求树的深度

/**************************
https://leetcode.com/problems/maximum-depth-of-binary-tree/
@date 2015.5.12
@description
给定一个二叉树，判断树的最大深度（也是树的高度）
按照题目的定义：树的最大深度指的是，从根节点到最远叶节点的途径中节点的个数。空树的深度为0

@solution
两种思路：
1.DFS 递归算出左右子树的深度，再取最大；
2.BFS 借助辅助队列，将同一层次的节点一起入队，在外的while循环统计层次数

***************************/

#include <iostream>
#include <queue>

using namespace std;

struct TreeNode{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x): val(x), left(NULL), right(NULL){}
} ;


class Solution{
public:
        int maxDepth(TreeNode *root){
        if (!root) return 0;

        int depth = 0; // 记录树深度
        queue<TreeNode *> temp; // 借助辅助队列
        temp.push(root);
        while(!temp.empty()){
            ++depth; // for循环体现BFS思想 把同一层的节点放在一个for循环里，循环外++depth
            for (int i = 0, n = temp.size(); i < n; i++){
                TreeNode *node = temp.front();
                temp.pop();
                if (node->left != NULL)
                    temp.push(node->left);
                if (node->right != NULL)
                    temp.push(node->right);
            }

        }

        return depth;
    }
};

TreeNode* insert(TreeNode *root, int data){
    TreeNode *ptr = root;
    TreeNode *tempNode;
    TreeNode *newNode = new TreeNode(data);

    if (ptr == NULL){
        return newNode;
    }else{
        while (ptr != NULL){
            tempNode = ptr;
            if (ptr->val <= data)
                ptr = ptr->right;
            else
                ptr = ptr->right;
        }
        if (tempNode->val <= data)
            tempNode->right = newNode;
        else
            tempNode->left = newNode;
    }
    return root;
}

int main(){
    TreeNode *root = NULL;
    int temp = 0;
    cin >> temp;
    while (temp != 0){
        root = insert(root, temp);
        cin >> temp;
    }

    Solution a;
    cout << a.maxDepth(root);
}