563. Binary Tree Tilt 二叉树的倾斜度

Given a binary tree, return the tilt of the whole tree.
The tilt of a tree node is defined as the absolute difference between the sum of all left subtree node values and the sum of all right subtree node values. Null node has tilt 0.
The tilt of the whole tree is defined as the sum of all nodes' tilt.
给定一二叉树,返回其整棵树的倾斜度。
倾斜度定义为左子树结点和与右子树结点和的绝对值之差。空节点的倾斜度为0。
整棵树的倾斜度定义为所有结点倾斜度的和。

Example:
Input:

         1
       /   \
      2     3

Output: 1
Explanation:
Tilt of node 2 : 0
Tilt of node 3 : 0
Tilt of node 1 : |2-3| = 1
Tilt of binary tree : 0 + 0 + 1 = 1

Note:

  1. The sum of node values in any subtree won't exceed the range of 32-bit integer.
  2. All the tilt values won't exceed the range of 32-bit integer.

思路
后序遍历,遍历过程中递归计算左右子树的结点和,同时使用全局变量记录当前节点倾斜度。

/**
 * 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 {
private:
    int res=0;
    int postOrder(TreeNode* root){
        if(!root) return 0;
        int left=postOrder(root->left);
        int right=postOrder(root->right);
        res+=abs(left-right);
        return left+right+root->val;
    }
public:
    int findTilt(TreeNode* root) {
        postOrder(root);
        return res;
    }
};

或不使用全局变量,改为参数传递。

/**
 * 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 {
private:
    int postOrder(TreeNode* root, int& res){
        if(!root) return 0;
        int left=postOrder(root->left,res);
        int right=postOrder(root->right,res);
        res+=abs(left-right);
        return left+right+root->val;
    }
public:
    int findTilt(TreeNode* root) {
        int res=0;
        postOrder(root,res);
        return res;
    }
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 目录 简书的 markdown 都不支持 [TOC] 语法……我就不贴目录了。下面按照类别,列出了29道关于二叉树...
    被称为L的男人阅读 3,349评论 0 8
  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,767评论 0 33
  • 人生如逆旅,我亦是行人。 须臾数十载,多少过客,谋面与否,都在自己的生命里留下了独特的印记。 简书上的每个你,多数...
    蜂蜜柚子与茶阅读 540评论 2 5
  • 中国历史上出了不少多才多艺的皇帝,除了大家熟知的建安时代的曹氏父子、南唐后主李煜、宋徽宗赵佶之外,南北朝时期的梁元...
    楚淑慧阅读 930评论 4 3
  • 岁月的童话,我生完孩子的第3年8个月 妈妈,咖啡去世了!! 午休起来,画画就从卧室的床上爬下...
    树村阅读 381评论 2 4