package main
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 * 
 * @param pRoot TreeNode类 
 * @return int整型
*/
func TreeDepth( pRoot *TreeNode ) int {
    if pRoot == nil {
        return 0
    }

    // write code here
    queue := []*TreeNode{pRoot}
    depth := 0

    for len(queue) > 0 {
        size := len(queue)

        for size > 0 {
            node := queue[0]
            queue = queue[1:]
            if node.Left != nil {
                queue = append(queue, node.Left)
            } 
            if node.Right != nil {
                queue = append(queue, node.Right)
            }
            size--
        }
        depth++
    }

    return depth
}