Medium
Sum of Consecutive Subarrays — Python
Full explanation · Time O(n) · Space O(1)
# Time: O(n)
# Space: O(1)
# combinatorics
class Solution(object):
def getSum(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
MOD = 10**9+7
def count(d):
result = total = l = 0
for i in xrange(len(nums)):
l += 1
total = (total+nums[i]*l)%MOD
result = (result+total)%MOD
if i+1 < len(nums) and nums[i+1]-nums[i] == d:
continue
total = l = 0
return result
return (count(1)+count(-1)-reduce(lambda accu, x: (accu+x)%MOD, nums, 0))%MOD