刷题平台:牛客网

题目

输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。

1.思路

可以是递归的方法,属于DFS(深度优先搜索);另一种方法是按照层次遍历,属于BFS(广度优先搜索)

  1. 5 TreeDepth(5)<br /> / \ / \<br /> 1 2 6 1 TreeDepath(1)<br /> \<br /> 7 TreeDepth(7)

2.代码

DFS

  1. private class TreeNode {
  2. int val;
  3. TreeNode left, right;
  4. TreeNode(int val) {
  5. this.val = val;
  6. }
  7. }
  8. public int TreeDepth(TreeNode root) {
  9. if (root == null) {
  10. return 0;
  11. }
  12. int left = TreeDepth(root.left);
  13. int right = TreeDepth(root.right);
  14. return Math.max(left, right) + 1;
  15. }
  1. class Solution {
  2. public:
  3. int depth(TreeNode *root) {
  4. if (root == nullptr) {
  5. return 0;
  6. }
  7. int left = depth(root->left);
  8. int right = depth(root->right);
  9. return max(left, right) + 1;
  10. }
  11. };

BFS

  1. import java.util.LinkedList;
  2. import java.util.Queue;
  3. /**
  4. * <p>Description: </p>
  5. *
  6. * @author yong.zhang
  7. * @version 1.0.0
  8. * @date 2020/4/30 8:18
  9. */
  10. public class Solution {
  11. private class TreeNode {
  12. int val;
  13. TreeNode left, right;
  14. TreeNode(int val) {
  15. this.val = val;
  16. }
  17. }
  18. public int TreeDepth(TreeNode root) {
  19. if (root == null) {
  20. return 0;
  21. }
  22. Queue<TreeNode> queue = new LinkedList<>();
  23. int depth = 0;
  24. queue.add(root);
  25. while (!queue.isEmpty()) {
  26. int size = queue.size();
  27. depth += 1;
  28. for (int i = 0; i < size; ++i) {
  29. TreeNode node = queue.poll();
  30. if (node.left != null) {
  31. queue.add(node.left);
  32. }
  33. if (node.right != null) {
  34. queue.add(node.right);
  35. }
  36. }
  37. }
  38. return depth;
  39. }
  40. }