Leetcode 1026：节点与其祖先之间的最大差值

题目描述

给定二叉树的根节点 root，找出存在于不同节点 A 和 B 之间的最大值 V，其中 V = |A.val - B.val|，且 A 是 B 的祖先。

（如果 A 的任何子节点之一为 B，或者 A 的任何子节点是 B 的祖先，那么我们认为 A 是 B 的祖先）

 

示例：

输入：[8,3,10,1,6,null,14,null,null,4,7,13]
输出：7
解释： 
我们有大量的节点与其祖先的差值，其中一些如下：
|8 - 3| = 5
|3 - 7| = 4
|8 - 1| = 7
|10 - 13| = 3
在所有可能的差值中，最大值 7 由 |8 - 1| = 7 得出。
 

提示：

树中的节点数在 2 到 5000 之间。
每个节点的值介于 0 到 100000 之间。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-difference-between-node-and-ancestor
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    void max_and_min(TreeNode* root,int &maxval,int &minval){
        if(root == NULL) return;
        maxval = max(maxval,root->val);
        minval = min(minval,root->val);
        max_and_min(root->left,maxval,minval);
        max_and_min(root->right,maxval,minval);
    }
    void dfs(TreeNode* root,int &ans){
        if(root == NULL) return;
        dfs(root->left,ans);
        dfs(root->right,ans);
        
        int tmp = 0;
        int maxval = INT_MIN,minval = INT_MAX;
        max_and_min(root->left,maxval,minval);
        if(maxval!=INT_MIN) tmp = max(tmp,abs(root->val-maxval));
        if(minval!=INT_MAX) tmp = max(tmp,abs(root->val-minval));
        maxval = INT_MIN,minval = INT_MAX;
        max_and_min(root->right,maxval,minval);
        if(maxval!=INT_MIN) tmp = max(tmp,abs(root->val-maxval));
        if(minval!=INT_MAX) tmp = max(tmp,abs(root->val-minval));
        ans = max(ans,tmp);
    }
    int maxAncestorDiff(TreeNode* root) {
        int ans = 0;
        dfs(root,ans);
        return ans;
    }
};