每日小细节新增算法好题合集,现在是初阶,每天会更新至少两道题目,感兴趣的小伙伴赶紧关注收藏不迷路哦
花费几分钟就能积累一种新的解题思路多是一件美事啊~
二叉树的搜索和之前的找到目标节点非常像,都是比较root->val和目标值,然后递归去找子树,如果恰好等于就返回节点,如果目标值比根节点的值还大,那就返回右子树,否则走左子树
struct TreeNode* searchBST(struct TreeNode* root, int val){
if(root==NULL)
{
return NULL;
}
else
{
if(root->val<val)
{
return searchBST(root->right,val);
}
else if(root->val>val)
{
return searchBST(root->left,val);
}
else
{
return root;
}
}
}
真是最简单的中等题....只要和刚才一样遍历,如果root==NULL,那就把root创建一个新节点然后把val放进去,左右节点都设为空
如果root->val < val 把右节点赋值为递归的右子树
如果root->val > val 把左节点赋值为递归的左子树
最后返回根节点
struct TreeNode* insertIntoBST(struct TreeNode* root, int val){
if(root==NULL)
{
root=(struct TreeNode* )malloc(sizeof(struct TreeNode));
root->val=val;
root->left=NULL;
root->right=NULL;
}
if(root->val<val)
{
root->right= insertIntoBST(root->right,val);
}
else if(root->val>val)
{
root->left= insertIntoBST(root->left,val);
}
return root;
}