563. 二叉树的坡度

题意描述：

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

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

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

示例:

注意：

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

 

题意理解：

利用先序遍历的思想，在遍历过程中为返回值加上每一个节点的坡点，最后返回即可。

不足：

存在多次节点的结点和重复计算，可以用hash表记录已经遍历过的路径，不过在本方法中没有实现，所以开销较大，但好在代码简单，容易理解。

 

代码：

#include <iostream>
#include <cmath>
using namespace std;


//树的结构
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
    int sum;
    int findTilt(TreeNode* root) {
        sum = 0;

        if(root)
        {
            PreOrder(root);
            return sum;
        }

        return sum;        
    }

    //遍历程序
    void PreOrder(TreeNode* root)
    {
        if(root)
        {
            sum += getTilt(root);       //遍历中将每个结点的坡度加到sum上
            PreOrder(root->left);
            PreOrder(root->right);
        }
    }

    // 获取一个点的坡度
    int getTilt(TreeNode* node)
    {
        if(node == nullptr)
            return 0;
        else 
        {
            int Left, Right;
            Left = Right = 0;
            if(node->left)                           //左子结点不为空，获取左子结点的数值和
                Left = getVal(node->left);
            if(node->right)                         //右子结点不为空，获取右子结点的数值和
                Right = getVal(node->right);        

            return abs(Left - Right);               //根据规则返回左右子节点的数值和差的绝对值
        }
    }

    //求一个节点以及所有子节点的数值和
    int getVal(TreeNode* node)
    {
        if(node == nullptr)
            return 0;
        else
            return node->val + getVal(node->left) + getVal(node->right);
    }

};

int main()
{
    TreeNode *root = new TreeNode(1);
    TreeNode *leftNode = new TreeNode(2);
    TreeNode *rightNode = new TreeNode(3);
    TreeNode *Node = new TreeNode(4);
    root->left = leftNode;
    leftNode->left = rightNode;
    leftNode->right = Node;

    Solution s;
    cout << s.findTilt(root) << endl;
    return 0;
}