938-二叉搜尋樹的範圍和

Null發表於2018-11-11

前言

Weekly Contest 110的第二題 二叉搜尋樹的範圍和：

給定二叉搜尋樹的根結點 root，返回 L 和 R（含）之間的所有結點的值的和。

二叉搜尋樹保證具有唯一的值。

返回日誌的最終順序

示例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

提示：

  1. 樹中的結點數量最多為 10000 個。
  2. 最終的答案保證小於 2^31。

解題思路

本題我選擇了一個笨方法去完成，就是利用二叉搜尋樹的一個特性：通過中序遍歷二叉搜尋樹後可以得到一個遞增的有序序列。所以我選擇了中序遍歷二叉搜尋樹獲得遞增且有序的陣列，然後遍歷陣列獲取需要的陣列元素進行累加。

實現程式碼

    /**
     * 938. 二叉搜尋樹的範圍和
     * @param root
     * @param L
     * @param R
     * @return
     */
    public int rangeSumBST(TreeNode root, int L, int R) {
        List<Integer> list=new ArrayList<Integer>();
        int result=0;
        if(root!=null){
            inorderTraversal(root,list);
        }
        for(int i:list){
            if(i>=L && i<=R){
                result+=i;
            }
        }
        return result;
    }

    /**
     * 中序遍歷(遞迴)
     * @param root
     * @param list
     */
    private void inorderTraversal(TreeNode root,List<Integer> list){
        if(root.left==null && root.right==null){
            list.add(root.val);
        }else if(root.left==null && root.right!=null){
            list.add(root.val);
            inorderTraversal(root.right,list);
        }else if(root.left!=null && root.right==null){
            inorderTraversal(root.left,list);
            list.add(root.val);
        }else{
            inorderTraversal(root.left,list);
            list.add(root.val);
            inorderTraversal(root.right,list);
        }
    }

相關文章