leetcode 307 区域和检索 - 数组可修改 Range Sum Query - Mutable

目录

1. AC 代码
   1. 解法一：sum函数(608 ms)
   2. 解法二：线段树(284 ms)

给定一个整数数组 nums，求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和，包含 i, j 两点。

update(i, val) 函数可以通过将下标为 i 的数值更新为 val，从而对数列进行修改。

示例:

Given nums = [1, 3, 5]

sumRange(0, 2) -> 9
update(1, 2)
sumRange(0, 2) -> 8
说明:

数组仅可以在 update 函数下进行修改。
你可以假设 update 函数与 sumRange 函数的调用次数是均匀分布的。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-sum-query-mutable

AC 代码

解法一：sum函数(608 ms)

直接循环求和过不了，但是使用 sum 函数可以过

class NumArray:
    
    def __init__(self, nums: List[int]):
        self.nums = nums
    
    def update(self, i: int, val: int) -> None:
        self.nums[i] = val

    def sumRange(self, i: int, j: int) -> int:
        return sum(self.nums[i:j+1])

解法二：线段树(284 ms)

class NumArray:
    
    def __init_tree(self, f, l, r):
        if l == r: return
        mid = (l + r) >> 1
        ls = f << 1
        rs = ls | 1
        self.__init_tree(ls, l, mid)
        self.__init_tree(rs, mid+1, r)
        self.t[f] = self.t[ls] + self.t[rs]


    def __init__(self, nums: List[int]):
        self.size = 1
        while self.size < len(nums): self.size <<= 1
        self.t = [0 for i in range(self.size<<1)]
        for i in range(len(nums)):
            self.t[self.size+i] = nums[i]
        self.__init_tree(1, 1, self.size)
    
    def __update(self, f, l, r):
        if l == r:
            if l == self.tar:
                self.t[f] = self.val
        else:
            mid = (l + r) >> 1
            ls = f << 1
            rs = ls | 1
            if self.tar > mid:
                self.__update(rs, mid+1, r)
            else:
                self.__update(ls, l, mid)
            self.t[f] = self.t[ls] + self.t[rs]
    
    def __query(self, f, l, r):
        if l == r:
            if self.ltar <= l <= self.rtar:
                return self.t[f]
            return 0
        mid = (l + r) >> 1
        ls = f << 1
        rs = ls | 1
        s = 0
        if self.rtar < l or self.ltar > r: return 0
        if self.ltar <=l and self.rtar >= r: return self.t[f]
        if self.rtar > mid:
            s += self.__query(rs, mid+1, r)
        if self.ltar <= mid:
            s += self.__query(ls, l, mid)
        return s
        
        
    def update(self, i: int, val: int) -> None:
        self.tar = i + 1
        self.val = val
        self.__update(1, 1, self.size)

    def sumRange(self, i: int, j: int) -> int:
        self.ltar = i + 1
        self.rtar = j + 1
        return self.__query(1, 1, self.size)