题目地址:
https://leetcode.com/problems/maximum-difference-between-node-and-ancestor/
给定一个二叉树,求最大的数 x x x使得树中存在两个节点 A A A和 B B B, A A A是 B B B的祖先,并且 x = ∣ A − B ∣ x=|A-B| x=∣A−B∣。
思路是DFS。DFS的时候,要向孩子节点传递当前节点及其祖先中的最大节点数值 M M M和最小节点数值 m m m,这样当遍历到节点 v v v的时候,用 ∣ v − M ∣ |v-M| ∣v−M∣和 ∣ v − m ∣ |v-m| ∣v−m∣更新最终答案即可。但需要注意,递归返回的时候要恢复现场。代码如下:
class Solution {
public:
int maxAncestorDiff(TreeNode* root) {
int res = 0;
dfs(root, 1e5 + 1, -1, res);
return res;
}
void dfs(TreeNode* cur, int minv, int maxv, int& res) {
if (!cur) return;
if (~maxv) {
res = max(abs(cur->val - minv), res);
res = max(abs(cur->val - maxv), res);
}
minv = min(minv, cur->val);
maxv = max(maxv, cur->val);
dfs(cur->left, minv, maxv, res);
dfs(cur->right, minv, maxv, res);
}
};
时间复杂度 O ( n ) O(n) O(n),空间 O ( h ) O(h) O(h)。