1. 程式人生 > >求二叉樹的深度和寬度[Java]

求二叉樹的深度和寬度[Java]

原文:http://www.cnblogs.com/xudong-bupt/p/4036190.html

這個是常見的對二叉樹的操作。總結一下:

設節點的資料結構,如下:

複製程式碼
1 class TreeNode {
2     char val;
3     TreeNode left = null;
4     TreeNode right = null;
5 
6     TreeNode(char _val) {
7         this.val = _val;
8     }
9 }
複製程式碼

 1.二叉樹深度

  這個可以使用遞迴,分別求出左子樹的深度、右子樹的深度,兩個深度的較大值+1即可。

複製程式碼
 1     // 獲取最大深度
 2     public static int getMaxDepth(TreeNode root) {
 3         if (root == null)
 4             return 0;
 5         else {
 6             int left = getMaxDepth(root.left);
 7             int right = getMaxDepth(root.right);
 8             return 1 + Math.max(left, right);
 9         }
10 }
複製程式碼

2.二叉樹寬度

  使用佇列,層次遍歷二叉樹。在上一層遍歷完成後,下一層的所有節點已經放到佇列中,此時佇列中的元素個數就是下一層的寬度。以此類推,依次遍歷下一層即可求出二叉樹的最大寬度。

複製程式碼
 1 // 獲取最大寬度
 2     public static int getMaxWidth(TreeNode root) {
 3         if (root == null)
 4             return 0;
 5 
 6         Queue<TreeNode> queue = new ArrayDeque<TreeNode>();
7 int maxWitdth = 1; // 最大寬度 8 queue.add(root); // 入隊 9 10 while (true) { 11 int len = queue.size(); // 當前層的節點個數 12 if (len == 0) 13 break; 14 while (len > 0) {// 如果當前層,還有節點 15 TreeNode t = queue.poll(); 16 len--; 17 if (t.left != null) 18 queue.add(t.left); // 下一層節點入隊 19 if (t.right != null) 20 queue.add(t.right);// 下一層節點入隊 21 } 22 maxWitdth = Math.max(maxWitdth, queue.size()); 23 } 24 return maxWitdth; 25 }
複製程式碼