文章目錄

• • 1. 題目
  • 2. 解題
  • • 遞歸+剪枝
    • 中序遍歷循環(huán)+剪枝

1. 題目

給定二叉搜索樹的根結(jié)點 root，返回 L 和 R（含）之間的所有結(jié)點的值的和。
題目的意思，節(jié)點的值在[L, R]這個區(qū)間內(nèi)，就加到結(jié)果里，求所有符合條件的節(jié)點值加和

示例 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提示：樹中的結(jié)點數(shù)量最多為 10000 個。 最終的答案保證小于 2^31。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/range-sum-of-bst
著作權(quán)歸領(lǐng)扣網(wǎng)絡所有。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請注明出處。

2. 解題

遞歸+剪枝

• 二叉搜索樹具有左子樹所有值小于根節(jié)點，右子樹大于根節(jié)點
• 根據(jù)以上性質(zhì)，注意遞歸法的剪枝

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);}} };

中序遍歷循環(huán)+剪枝

• 中序是遞增序列，當當前值大于R時，后面均大于R，break

運行時間比較長，可能減的枝子沒有遞歸多，遞歸枝杈比較多，隨著深度變大，很大程度可被減掉

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)//中序非降，后面都大于Rbreak;root = root->right;}return sum;} };

總結(jié)

以上是生活随笔為你收集整理的LeetCode 938. 二叉搜索树的范围和（二叉树遍历+搜索剪枝）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。