Lisa's_Workbook.py

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the workbook function below.
def workbook(n, k, arr):
    answer = 0
    page = 1
    for problems in arr:
        for index in range(1,problems+1):
            if index == page:
                answer +=1
            
            if index == problems or index %k == 0:
                page+=1
    return answer


if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    nk = input().split()

    n = int(nk[0])

    k = int(nk[1])

    arr = list(map(int, input().rstrip().split()))

    result = workbook(n, k, arr)

    fptr.write(str(result) + '\n')

    fptr.close()