线段树查询 II

题目描述：对于一个数组，我们可以对其建立一棵 线段树, 每个结点存储一个额外的值 count 来代表这个结点所指代的数组区间内的元素个数. (数组中并不一定每个位置上都有元素) 

实现一个 query 的方法，该方法接受三个参数 root, start 和 end, 分别代表线段树的根节点和需要查询的区间，找到数组中在区间[start, end]内的元素个数。

样例：

对于数组 [0, 空，2, 3], 对应的线段树为：

query(1, 1), return 0

query(1, 2), return 1

query(2, 3), return 2

query(0, 2), return 2

跟“线段树的查询是一样的”（详见：点击打开链接），只不过此处求的是count，而那一道题求的是max，所以，很容易先想到最后对于count的处理时+，而不是对max的处理那样用max()函数。

除此之外，基本也就没什么不同了，对了，还有一点是当查询区间与节点区间不交时，返回0，而不是空，这一点与上题不同。

我们仿照上一题的思路写出代码：因为和上一题的思路太一致，所以，不再赘述

"""
Definition of SegmentTreeNode:
class SegmentTreeNode:
    def __init__(self, start, end, count):
        self.start, self.end, self.count = start, end, count
        self.left, self.right = None, None
"""

class Solution: 
    # @param root, start, end: The root of segment tree and 
    #                          an segment / interval
    # @return: The count number in the interval [start, end] 
    def query(self, root, start, end):
        if root is None or start > root.end or end < root.start:
            return 0
        elif start <= root.start and end >= root.end:
            return root.count
        else:
            return self.query(root.left, start, end) + self.query(root.right, start, end)
        # write your code here