節點數據結構node
class TreeNode { TreeNode left = null; TreeNode right = null; }
最大深度,基本思路是:使用遞歸,分別求出左子樹的深度、右子樹的深度,兩個深度的較大值+1就是最大深度。數據結構
// 獲取最大深度 public static int getMaxDepth(TreeNode treeNode) { if (treeNode == null) return 0; else { int left = getMaxDepth(treeNode.left); int right = getMaxDepth(treeNode.right); return 1 + Math.max(left, right); } }
// 獲取最小深度 public static int getMinDepth(TreeNode treeNode) { if (treeNode == null) return 0; else { int left = getMinDepth(treeNode.left); int right = getMinDepth(treeNode.right); return 1 + Math.min(left, right); } }
最大寬度,基本思路:使用隊列,按層次遍歷二叉樹。在上一層遍歷完成後,下一層的全部節點已經放到隊列中,此時隊列中的元素個數就是下一層的寬度。以此類推,依次遍歷下一層便可求出二叉樹的最大寬度。spa
// 獲取最大寬度 public static int getMaxWidth(TreeNode treeNode) { if (treeNode == null) return 0; Queue<TreeNode> queue = new ArrayDeque<TreeNode>(); int maxWitdth = 1; // 最大寬度 queue.add(treeNode); // 入隊 while (true) { int len = queue.size(); // 當前層的節點個數 if (len == 0) break; while (len > 0) {// 若是當前層,還有節點 TreeNode node = queue.poll(); len--; if (node.left != null) queue.add(node.left); // 下一層節點入隊 if (node.right != null) queue.add(node.right);// 下一層節點入隊 } maxWitdth = Math.max(maxWitdth, queue.size()); } return maxWitdth; }
推薦一個在線寫白板代碼的好地方:http://www.lintcode.com/ code
在線能力評估:https://www.nowcoder.com/blog