算法笔记
二叉搜索树的问题是二叉树当中必须要理解且要会解决的问题,二叉搜索树的搜索既可以使用递归也可以使用非递归的方式去解决,都比较容易理解。
题目描述
解题思路
二叉搜索树可以看做是一个有序数组,搜索节点时一般只需要判断节点值与目标值进行比较,如果节点值大于目标值,就在左子树里面寻找,同理小于的情况也是一样的。
递归方法里面,我们需要定义一个result去记录以目标值为根节点的子树将满足条件的子树装入树里面即可。
非递归方法里面,同理只需要把递归的过程变为循环的过程即可。
代码部分
递归
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
TreeNode* result = new TreeNode(0);
if(root == nullptr || root->val == val) {
return root;
}
if(root->val > val) {
result = searchBST(root->left, val);
}else if(root->val < val) {
result = searchBST(root->right, val);
}
return result;
}
};
非递归
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
if(root == nullptr) {
return root;
}
while(root) {
if(root->val > val) {
root = root->left;
}else if(root->val < val) {
root = root->right;
}else {
return root;
}
}
return nullptr;
}
};
结语
二叉树搜索树的题目解决起来都会与其特性有所关联,因此在解题时可以优先思考其特性来求解相关题目。