python数组部分求和_Python3按指定数组坐标index求和

Input：

arr -- list of int, for example [0,1,2,3,4,5]

summary_index --list of index range, for example[(0,2),(3,4),(0,5)]

Output:

result -- list of sum of arr of the index range in summary_index, for example[3, 7,15]

传统写法：只需要一条sum函数结合列表就能完成优点：代码简洁，只需要一行

缺点：算法复杂是O(n2) (说明：python自带的sum()函数实现就有for循环)

>>> arr=[0,1,2,3,4,5]

>>> summary_index=[(0,2),(3,4),(0,5)]

>>> result=[sum(arr[item[0]: item[1]+1]) for item in summary_index]

>>> print(result)

[3, 7, 15]

进阶写法：降低算法复杂度，大思路只能是空间换时间优点：算法复杂度降到O(n)

缺点：无，多写点代码又算什么呢？

思路：

1) 先生成一个数组arr_sum，元素是arr前n项的和(n<=len(arr))

2) arr数组x到y的和，如果x等于0，和等于arr_sum[y];如果x大于0，和等于arr_sum[y]-arr_sum[x-1]

>>> arr=[0,1,2,3,4,5]

>>> summary_index=[(0,2),(3,4),(0,5)]

>>> for i,value in enumerate(arr):

... sum=value if i==0 else arr_sum[i-1]+value

... arr_sum.append(sum)

...

>>> print(arr_sum)

[0, 1, 3, 6, 10, 15]

>>> result2=[]

>>> for item_sum in summary_index:

... value=arr_sum[item_sum[1]]-arr_sum[item_sum[0]-1] if item_sum[0]>0 else arr_sum[item_sum[1]]

... result2.append(value)

...

>>> print(result2)

[3, 7, 15]