前往小程序,Get更优阅读体验!
立即前往
发布
社区首页 >专栏 >LeetCode 938. 二叉搜索树的范围和(二叉树遍历+搜索剪枝)

LeetCode 938. 二叉搜索树的范围和(二叉树遍历+搜索剪枝)

作者头像
Michael阿明
发布2021-02-20 14:27:34
发布2021-02-20 14:27:34
27100
代码可运行
举报
运行总次数:0
代码可运行

1. 题目

给定二叉搜索树的根结点 root,返回 L 和 R(含)之间的所有结点的值的和。 题目的意思,节点的值在[L, R]这个区间内,就加到结果里,求所有符合条件的节点值加和

代码语言:javascript
代码运行次数:0
复制
示例 1:

输入:root = [10,5,15,3,7,null,18], L = 7, R = 15
输出:32
示例 2:

输入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出:23
 

提示:

树中的结点数量最多为 10000 个。
最终的答案保证小于 2^31。

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/range-sum-of-bst 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

递归+剪枝

  • 二叉搜索树具有左子树所有值小于根节点,右子树大于根节点
  • 根据以上性质,注意递归法的剪枝
代码语言:javascript
代码运行次数:0
复制
class Solution {
public:
    int rangeSumBST(TreeNode* root, int L, int R) {
        int sum = 0;
        sumLR(root,L,R,sum);
        return sum;
    }
    void sumLR(TreeNode* root,int &L,int &R,int &sum)
    {
    	if(root == NULL)
    		return;
    	if(root->val >= L && root->val <= R)
		{
			sum += root->val;
			sumLR(root->left,L,R,sum);
    		sumLR(root->right,L,R,sum);
		}
    	else if(root->val < L)
    	{//左子树都小于L,砍了
    		sumLR(root->right,L,R,sum);
    	}
    	else if(root->val > R)
    	{//右子树都大于R,砍了
    		sumLR(root->left,L,R,sum);
    	}
    }
};

中序遍历循环+剪枝

  • 中序是递增序列,当当前值大于R时,后面均大于R,break

运行时间比较长,可能减的枝子没有递归多,递归枝杈比较多,随着深度变大,很大程度可被减掉

代码语言:javascript
代码运行次数:0
复制
class Solution {
public:
    int rangeSumBST(TreeNode* root, int L, int R) {
        int sum = 0;
        stack<TreeNode*> stk;
        while(root || !stk.empty())
        {
        	while(root)
        	{
        		stk.push(root);
        		root = root->left;
        	}
        	root = stk.top();
        	stk.pop();
        	if(root->val >= L && root->val <= R)
        		sum += root->val;
        	else if(root->val > R)//中序非降,后面都大于R
        		break;
        	root = root->right;
        }
        return sum;
    }
};
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019/09/07 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
    • 递归+剪枝
    • 中序遍历循环+剪枝
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档