leetcode303-range sum query-python

Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.
给定一个整形数组,求出元素i,j之间所有元素的和。
Example:
Given nums = [-2, 0, 3, -5, 2, -1]

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3
Note:
You may assume that the array does not change.
There are many calls to sumRange function.

代码如下:

class NumArray(object):
    def __init__(self,num):
        self.num=num

    def sumRange(self,i,j):
        sum=0
        while(i<=j):
            sum+=self.num[i]
            i=i+1
        print(sum)

num1=NumArray([-2,0,3,-5,2,-1])

num1.sumRange(1,2)
num1.sumRange(0,4)

第一次做leetcode题,测试结果超时,还未解决。
问题是题目说多次调用sumRange函数,时间复杂度o(n*n)
利用数组保存每一个list不同长度部分的和,计算时直接调用数组的值就可以,用空间换时间。

class NumArray(object):
    def __init__(self, nums):
        """
        initialize your data structure here.
        :type nums: List[int]
        """
        size = len(nums)
        self.sums = [0] * (size + 1)
        for x in range(size):
            self.sums[x + 1] = self.sums[x] + nums[x]

    def sumRange(self, i, j):
        """
        sum of elements nums[i..j], inclusive.
        :type i: int
        :type j: int
        :rtype: int
        """
        return self.sums[j + 1] - self.sums[i]

self.sums = [0] * (size + 1)这一行的意思是给sums数组初始化。如果size是5,表示代求数组有5个元素。则sums数组有6个元素,其中sums[0]=0。其他5个值在下面被赋值,分别是每一部分的和。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值