剑指offer - 面试题55扩展: 平衡二叉树 - C++

面试题55非扩展:二叉树深度,用递归很容易就写出来了,实际上是前序遍历。

扩展题平衡二叉树,我的刚开始的思路也是从上到下求左右子树深度判断当前节点是否平衡&&递归左右子树是否平衡。但是自己隐约觉得这样重复遍历了子节点,应该从下向上只遍历一次。看书上果然是这个思路。书上指出这其实是二叉树的后序遍历,由于我自己的不熟练,导致并没有意识到这是在考后序遍历。思考一:去掌握二叉树的前中后序遍历的循环、递归写法。基本功这个东西,只要偷懒必将体现出来。

参考了书上的代码后,一开始觉得看不懂,自己用例子过一遍代码后有点明白了,觉得逻辑确实有点点复杂,另外发现书上的代码不足够的精致,导致了一点理解上的小坎儿。

参照书上的思路自己写的代码:

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        int depth;
        return IsBalanced(pRoot, &depth);
    }
    
    bool IsBalanced(TreeNode* pRoot, int* depth) {
        if(pRoot == nullptr) {
            *depth = 0;
            return true;
        }
        int left, right;
        if(IsBalanced(pRoot->left, &left) 
           && IsBalanced(pRoot->right, &right)) {
            int diff = left-right;
            if(diff>=-1 && diff<=1) {
                *depth = 1 + (left > right ? left : right);
                return true;
            }
        }
        return false;
    }
};

所得:

参数用指针,传参加引用这种方试让一个变量函数内外共享。这点我隐约意识到了,但是之前都用的是数组,没有意识到指针可以是任何类型。

希望再用一周把剑指第一遍刷完,意味着刷题所需基础都学过了;之后看情况开刷leetcode(听说discuss区讲的也很好),把剑指的没掌握的题再刷,第二遍最好用java;多看讨论区,可能有比书上好的解法;把用到的技巧&思路&方法都总结。毕竟总结和温习的性价比是非常高的~

猜你喜欢

转载自blog.csdn.net/L_bic/article/details/87889749