(刷题笔记)Leetcode 563.二叉树的坡度

目录

  • 题目
  • 解题思路
  • 代码(C++)

题目

给定一个二叉树,计算整个树的坡度。

一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。

整个树的坡度就是其所有节点的坡度之和。

示例:

输入:

     1
   /   \
  2     3
输出:1
解释:
结点 2 的坡度: 0
结点 3 的坡度: 0
结点 1 的坡度: |2-3| = 1
树的坡度 : 0 + 0 + 1 = 1

提示:

任何子树的结点的和不会超过 32 位整数的范围。
坡度的值不会超过 32 位整数的范围。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-tilt

解题思路

最近写了好多二叉树的题目,但是碰到这样的还是不会写,分析了一下觉得应该是后序遍历,但是后序的迭代不会写,递归也分析不来,最后还是得去看答案。
一开始看官方给的答案不是很懂,然后看了评论区的一个答案,那个答案的方式有点类似二叉树的最大深度的解法,
就是维护一个递归函数,可以求每个结点所有子树加本身结点的和,然后在主函数里再用递归对每个结点遍历,并利用前面的递归函数求其左右子树的和,并求坡度。
不过这个方法对结点的遍历次数太多了,效率很低。
然后就是官方写的解法,递归版本的后序遍历,执行用时果然少了很多。不过还是不懂后序遍历怎么用迭代写,我看后面提交记录里有人用迭代解这一道题,但是又用了stack又用了set,
感觉挺复杂的,目前我功力还不够。
总是在找工作的时候才发现自己啥都不会,实在是太难了,现在才开始学习语言和数据结构也不知道来不来的及。

代码(C++)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
     
public:


    int dfs(TreeNode* root,int & tilt){
     

        if(root==NULL) return 0;
        int left=dfs(root->left,tilt);
        int right=dfs(root->right,tilt);

        tilt+=abs(left-right);

        return left+right+root->val;

    }

    
    int findTilt(TreeNode* root) {
     
        int tilt=0;
        dfs(root,tilt);
        return tilt;

    }
};

你可能感兴趣的:(刷题笔记,leetcode,二叉树,c++)