每日一題:1026. 節點與其祖先之間的最大差值

SandaiYoung發表於2024-04-05

給定二叉樹的根節點 root,找出存在於 不同 節點 AB 之間的最大值 V,其中 V = |A.val - B.val|,且 AB 的祖先。

(如果 A 的任何子節點之一為 B,或者 A 的任何子節點是 B 的祖先,那麼我們認為 A 是 B 的祖先)

示例 1:

輸入:root = [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 得出。

我們不需要維護每個節點的最大差值,只需要維護一條路徑上的最大元素和最小元素就夠了,外加一個深度優先遍歷。

class Solution {
public:
    int ans=0;
    void dfs(TreeNode* node,int mn,int mx){
        if(node==nullptr)return ;
        mn=min(mn,node->val);
        mx=max(mx,node->val);
        ans=max(ans,max(node->val-mn,mx-node->val));
        dfs(node->right,mn,mx);
        dfs(node->left,mn,mx);
    }
    int maxAncestorDiff(TreeNode* root) {
        dfs(root,root->val,root->val);
        return ans;
    }
};

相關文章