【leetcode刷题】T143-二叉树的坡度

木又连续日更第100天(100/100)


木又的第143篇leetcode解题报告

二叉树类型第33篇解题报告

leetcode第563题:二叉树的坡度

https://leetcode-cn.com/problems/binary-tree-tilt


【题目】

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

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

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

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

注意:

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

【思路】

分别得到左右子树的元素和,再计算并累加坡度。

有些小伙伴或许还在疑问,递归到底怎么写呀?

其实很简单的。

主要两步:假设已经有个函数能够实现我们想要的子功能,我们要怎样调用;函数应该什么时候返回结果,返回什么结果。

比如这道题

1)假设我们有个函数能够实现求子树和的功能,那么怎么调用求得坡度呢?

# 求得左右子树和
left = fun(node.left)
right = fun(node.right)
# 求得坡度
tilt = abs(left - right)

2)什么时候返回子树和呢?

一是访问到叶子节点,返回0;二是函数结束,返回以本节点为根节点的树的和。

if not node:
    return 0
left = ...
...
return left + right + node.val

最后把两段代码合在一起,解决几个问题:哪些应该是全局变量,是否还有特殊情况等。

【代码】

python版本

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def findTilt(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        self.tilt = 0
        self.get_tilt(root)
        return self.tilt

    def get_tilt(self, node):
        if not node:
            return 0
        left = self.get_tilt(node.left)
        right = self.get_tilt(node.right)
        print(node.val, left, right)
        self.tilt += abs(left - right)
        return left + right + node.val

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 findTilt(TreeNode* root) {
        int tilt = 0;
        get_tilt(root, tilt);
        return tilt;
    }

    int get_tilt(TreeNode* node, int& tilt){
        if(!node)
            return 0;
        int left = get_tilt(node->left, tilt);
        int right = get_tilt(node->right, tilt);
        tilt += abs(left - right);
        return left + right + node->val;
    }
};

原文发布于微信公众号 - 木又AI帮(gh_eaa31cab4b91)

原文发表时间:2019-08-15

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券