Jumping_on_the_Clouds_Revisited.py

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the jumpingOnClouds function below.
def jumpingOnClouds(c, k):
    e = 100
    start = 0
    while True:
        i = start
        i = (i + k)%(len(c))
        e = e - 1
        if c[i] == 1:
            e = e - 2
        start = i
        if start == 0:
            break
    
    return e

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

    nk = input().split()

    n = int(nk[0])

    k = int(nk[1])

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

    result = jumpingOnClouds(c, k)

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

    fptr.close()