給定二叉樹的根節點 root
,找出存在於 不同 節點 A
和 B
之間的最大值 V
,其中 V = |A.val - B.val|
,且 A
是 B
的祖先。
(如果 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;
}
};