题目描述
解法
这道题在一个结点处要同时判断三件事情:
- 左右子树是否是 BST
- 左子树的最大值和右子树的最小值
- 左右子树的节点值之和
无疑,采用后序遍历才能知道该结点左右子树的情况。同时,因为同时要判断的内容比较多,我们用一个数组来存储一下这些信息,记这个数据为 i n f o s infos infos,长度为 4 4 4
- i n f o s [ 0 ] infos[0] infos[0] 记录以 root 为根的二叉树是否是 BST,若为 1 1 1 则说明是 BST,若为 0 0 0 则说明不是 BST
- i n f o s [ 1 ] infos[1] infos[1] 记录以 root 为根的二叉树所有节点中的最小值
- i n f o s [ 2 ] infos[2] infos[2] 记录以 root 为根的二叉树所有节点中的最大值
- i n f o s [ 3 ] infos[3] infos[3] 记录以 root 为根的二叉树所有节点值之和
如此再看下面的代码就非常好理解了
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int maxSumBST(TreeNode* root) {
int res = 0;
traverse(root, res);
return res;
}
vector<int> traverse(TreeNode* root, int& mmax){
if (root == nullptr) return {1, INT_MAX, INT_MIN, 0};
vector<int> left = traverse(root->left, mmax);
vector<int> right = traverse(root->right, mmax);
vector<int> infos = vector<int>(4, 0);
if (left[0] && right[0] && root->val > left[2] && root->val < right[1])
{
infos[0] = 1;
infos[1] = min(left[1], root->val);
infos[2] = max(right[2], root->val);
infos[3] = root->val + left[3] + right[3];
mmax = max(mmax, infos[3]);
}
else infos[0] = 0;
return infos;
}
};