http://oj.leetcode.com/problems/maximum-depth-of-binary-tree/
这是一道求 二叉树深度的题,一开始用递归的思路有严重的问题: 单纯为了递归二递归,没有把问题分解成可以递归的子问题。
一开始一直不能AC的做法是: 递归遍历所有的路径长度,然后在ector中取最大值。
参考了:http://www.cnblogs.com/GoAhead/archive/2012/05/22/2513847.html 问题就变成了:
为了求得树的深度,可以先求左右子树的深度,取二者较大者加1即是树的深度,递归返回的条件是若节点为空,返回0。
另外在这篇博文里也写了一个非递归算法,
非递归实现基本思想:
受后续遍历二叉树思想的启发,想到可以利用后续遍历的方法来求二叉树的深度,在每一次输出的地方替换成算栈S的大小,遍历结束后最大的栈S长度即是栈的深度。
算法的执行步骤如下:
(1)当树非空时,将指针p指向根节点,p为当前节点指针。
(2)将p压入栈S中,0压入栈tag中,并令p执行其左孩子。
(3)重复步骤(2),直到p为空。
(4)如果tag栈中的栈顶元素为1,跳至步骤(6)。从右子树返回
(5)如果tag栈中的栈顶元素为0,跳至步骤(7)。从左子树返回
(6)比较treedeep与栈的深度,取较大的赋给treedeep,对栈S和栈tag出栈操作,p指向NULL,并跳至步骤(8)。
(7)将p指向栈S栈顶元素的右孩子,弹出栈tag,并把1压入栈tag。(另外一种方法,直接修改栈tag栈顶的值为1也可以)
(8)循环(2)~(7),直到栈为空并且p为空
(9)返回treedeep,结束遍历
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
vector
his;
class Solution {
public:
int maxDepth(TreeNode *root) {
int deep = 0;
if(root != NULL) {
int leftTreeDepth = maxDepth(root->left);
int rightTreeDepth = maxDepth(root->right);
deep = (leftTreeDepth >= rightTreeDepth ? leftTreeDepth : rightTreeDepth) + 1;
}
return deep;
}
};