前缀和与差分——前缀和O(n)

p5_1_前缀和与差分——前缀和O(n)
题目
输入一个长度为 n 的整数序列。

接下来再输入 m 个询问,每个询问输入一对 l,r。

对于每个询问,输出原序列中从第 l 个数到第 r 个数的和。

输入格式
第一行包含两个整数 n 和 m。

第二行包含 n 个整数,表示整数数列。

接下来 m 行,每行包含两个整数 l 和 r,表示一个询问的区间范围。

输出格式
共 m 行,每行输出一个询问的结果。

数据范围
1≤l≤r≤n,
1≤n,m≤100000,
−1000≤数列中元素的值≤1000
输入样例:
5 3
2 1 3 6 4
1 2
1 3
2 4
输出样例:
3
6
10

#a1 a2 a3 ... an
#前缀和Si=a1+a2+...+ai (下标要从1开始,就可以定义S0=0)

#计算 S[i]=s[i-1]+ai这样递归写!!!!这样easy   和等差数列的和有点像
#作用:若想求[l,r]这一段数的和,也就是al+a(l+1)+...+ar,若没有前缀和得循环一次时间O(n),有前缀和就=Sr-S(l-1) 这样时间O(1) 和等差数列的和有点像

#(下标要从1开始,就可以定义S0=0,且求[1,10]段=S10-S0,需要用到S0 这样定义S0可以保证[l,r]求解公式一致,少一个判断条件

def PrefixSum(A,l,r):
    S=[0]#即S[0]=0
    for x in A:
        S.append(S[-1]+x)#目标是S[i]=S[i-1]+A[i],但这里要注意不能for i in range(len(A)),因为A的下标是从0开始的和前面论述A下标从1开始不行,所以这里不能用下标做,相应的S也是
    # S=[0]*(n+10)
    # for i in range(n):
    #     S[i+1]=S[i]+A[i]  #这样写也行,要注意初始S就得设定值才能这样,并且要注意是A[i]不是A[i+1]
    return S[r]-S[l-1]



if __name__=="__main__":
    n,m=map(int,input().split())
    A=list(map(int,input().split()))#这里得split 因为ai不一定属于0-9之间
    for i in range(m):
        l,r=map(int,input().split())
        print(PrefixSum(A,l,r))

上面这个会超时

#前面超时的原因是写了函数,一旦写了函数,每次输如(l,r)都会重新算Si,导致超时
if __name__=="__main__":
    n,m=map(int,input().split())
    A=list(map(int,input().split()))#这里得split 因为ai不一定属于0-9之间
    S = [0]
    for x in A:
        S.append(S[-1] + x)

    for i in range(m):
        l,r=map(int,input().split())
        print(S[r]-S[l-1])





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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值