树的遍历巧用
当判断节点之间的关系时,第一个要想到的就是以哪种顺序的遍历方式可以更好得到结果
如下,验二叉搜索树的时候,中序遍历,如果满足递增,则刚好符合条件,因此使用中序遍历
/*
98. 验证二叉搜索树
给定一个二叉树,判断其是否是一个有效的二叉搜索树。
假设一个二叉搜索树具有如下特征:
节点的左子树只包含小于当前节点的数。
节点的右子树只包含大于当前节点的数。
所有左子树和右子树自身必须也是二叉搜索树。
示例 1:
输入:
2
/ \
1 3
输出: true
示例 2:
输入:
5
/ \
1 4
/ \
3 6
输出: false
解释: 输入为: [5,1,4,null,null,3,6]。
根节点的值为 5 ,但是其右子节点值为 4
*/
class Solution {
public:
int flag=0;
long vals=-2147483649;
void dfs(TreeNode* root){
if(root){
dfs(root->left);
if(root->val<=vals){
flag=1;
}else{
vals=root->val;
}
dfs(root->right);
}
}
bool isValidBST(TreeNode* root) {
dfs(root);
if(flag==1){
return false;
}else{
return true;
}
}
};
恢复二叉搜索树
理解:把节点当作一个下标,和上面的区别就是实例化三个新下标,一个用来判断顺序增加(该哨兵大于root则不顺序,然后设置新的下标即可)
/**
99. 恢复二叉搜索树
给你二叉搜索树的根节点 root ,该树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。
进阶:使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗?
示例 1:
输入:root = [1,3,null,null,2]
输出:[3,1,null,null,2]
解释:3 不能是 1 左孩子,因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。
示例 2:
输入:root = [3,1,4,null,null,2]
输出:[2,1,4,null,null,3]
解释:2 不能在 3 的右子树中,因为 2 < 3 。交换 2 和 3 使二叉搜索树有效。 */
class Solution {
public:
TreeNode* ans=new TreeNode(-2147483648);//记录递增的哨兵节点
TreeNode* first=nullptr;//记录第一个逆序
TreeNode* next=nullptr;//记录第二个逆序
void inorder(TreeNode* root)
{
if(!root){
return;
}
inorder(root->left);
if (first== nullptr && ans->val > root->val) first = ans;
if (first != nullptr && ans->val > root->val) next = root;
ans=root;
inorder(root->right);
}
void recoverTree(TreeNode* root) {
inorder(root);
int temp=first->val;
first->val=next->val;
next->val=temp;
}
};