Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- n, m = map(int, input().split())
- arr = list(map(int, input().split()))
- for i in range(1, n):
- if arr[i]:
- if i - m > 0:
- arr[i] = 2*arr[i-1] - arr[i-m-1]
- else:
- arr[i] -= 1
- for jump in range(1, m + 1):
- if i - jump >= 0:
- arr[i] += arr[i - jump]
- else:
- arr[i] = arr[i-1]
- arr[i] %= 1000000007
- print(arr.pop())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement