leetcode之Range Sum Query - Mutable

原创于 2015-11-19 01:32:05 发布 · 447 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode #Range Sum Query - Mu

本文介绍了一种不使用复杂的数据结构线段树来实现数组的更新与查询功能的方法。通过简单的数组操作实现了对指定区间元素求和及更新单个元素的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这次用的还是土方法，没有用高大上的线段树，竟然还是过了。。。

class NumArray(object):
    def __init__(self, nums):
        """
        initialize your data structure here.
        :type nums: List[int]
        """
        self.items = nums
        self.sum1 = sum(self.items)
        

    def update(self, i, val):
        """
        :type i: int
        :type val: int
        :rtype: int
        """
        self.sum1 = self.sum1 - self.items[i] + val
        self.items[i] = val
        

    def sumRange(self, i, j):
        """
        sum of elements nums[i..j], inclusive.
        :type i: int
        :type j: int
        :rtype: int
        """
        a = 0
        if j - i >= len(self.items) / 2:
            return self.sum1 - sum(self.items[:i]) - sum(self.items[j + 1:])
        else:
            return sum(self.items[i: j + 1])