563. Binary Tree Tilt

/**
 * 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 tiltIter(TreeNode* root, int *t) {
        if (root == nullptr) {
            return 0;
        }
        
        int leftSum = 0;
        int rightSum = 0;
        if (root->left != nullptr) {
            leftSum = tiltIter(root->left, t);
        }
        if (root->right != nullptr) {
            rightSum = tiltIter(root->right, t);
        }
        //tilt of the current node;
        int tilt = abs(leftSum - rightSum);
        *t += tilt;
        return root->val + leftSum + rightSum;
    }
    
    int findTilt(TreeNode* root) {
        int tilt = 0;
        tiltIter(root, &tilt);
        return tilt;
    }
};