1. 程式人生 > >二叉樹最大深度(Java)

二叉樹最大深度(Java)

題目:輸入一棵二叉樹的根節點,求該樹的深度。從根節點到葉子結點一次經過的結點形成樹的一條路徑,最長路徑的長度為樹的深度。根節點的深度為1。
解體思路:
如果根節點為空,則深度為0,返回0,遞迴的出口
如果根節點不為空,那麼深度至少為1,然後我們求他們左右子樹的深度,
比較左右子樹深度值,返回較大的那一個
通過遞迴呼叫

public class BinaryTreeDepth{public static void main(String[] args){//建立二叉樹


    TreeNode root = new TreeNode(1);TreeNode[] node = new TreeNode[6];for(int i=0; i<6; i++){node[i] = new TreeNode(i+2);System.out.println(node[i].value+" ");}root.leftNode = node[0];root.rightNode = node[1];node[0].leftNode = null;node[0].rightNode = node[3];node[1].leftNode = node[4];node[1].leftNode = node[5];
    //計算二叉樹深度
    BinaryTreeDepth btd = new BinaryTreeDepth();int num = btd.depthOfTree( root);System.out.println("The depth of binaryTree is:"+num);}private int depthOfTree(TreeNode root){
    //遞迴結束條件
    if(root == null){return 0;}
    //根節點不為空時加1
    if(root.leftNode == null){return depthOfTree(root.rightNode)+1;}if(root.rightNode == null){return depthOfTree(root.leftNode)+1;}
    //左右子節點都不為空,返回左右子樹的最大深度加1
    return MaxDepth(depthOfTree(root.leftNode), depthOfTree(root.rightNode))+1;}private int MaxDepth(int a, int b){return a>b?a:b;}}
    }