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])