要求:求二叉树的深度(二叉树的深度为最远叶子节点到根节点的距离,即根节点到最远叶子节点的距离)
Given a binary tree, find its maximum depth. The maximum depth is the number of nodes along the longest path from the root node down to the farthest leaf node.
有两种求解的思路,一种采用DFS的思想,一种采用BFS的思想,如下代码所示:
1 struct TreeNode { 2 int val; 3 TreeNode* left; 4 TreeNode* right; 5 TreeNode(int x): val(x), left(NULL),right(NULL) {} 6 }; 7 8 //采用DFS的思想 9 int maxDepth(TreeNode *root) 10 { 11 if (NULL == root) 12 return 0; 13 int l = maxDepth(root->left); 14 int r = maxDepth(root->right); 15 16 return l > r ? l + 1:r+1; 17 //以上这两种方式有一种更简便的方法 18 //return 1 + max(maxDepth(root->left), maxDepth(root->right)); 19 } 20 21 //采用BFS的方法,引入队列 22 int maxDepth(TreeNode *root) 23 { 24 if (NULL == root) 25 return 0; 26 queue <TreeNode *> que; 27 int nCount = 1; 28 int nDepth = 0;// 记录队列里面每一层上的元素 29 30 que.push(root); 31 while(!que.empty()) { 32 TreeNode *pTemp = que.front(); 33 que.pop(); 34 nCount --; 35 36 if (pTemp->left) 37 que.push(pTemp->left); 38 if (pTemp->right) 39 que.push(pTemp->right); 40 41 if (nCount == 0) { 42 nDepth ++; 43 nCount = que.size(); 44 } 45 } 46 return nDepth; 47 }