Problem p03074 (python)

Fixed submission:
from itertools import accumulate

n, k = list(map(int, input().split()))
s = input()


def solve():
    l = []
    prev = '1'
    cnt = 0
    for c in s:
        if prev == c:
            cnt += 1
        else:
            l.append(cnt)
            cnt = 1
        prev = c
    l.append(cnt)
    if prev == '0':
        l.append(0)

    w = 2 * k + 1

    if len(l) <= w:
        return n

    acc = [0] + list(accumulate(l))
    ans = 0
    for i in range(0, len(l) - w + 1, 2):
        ans = max(ans, acc[i + w] - acc[i])
    return ans


print(solve())

Buggy submission (verdict: Runtime Error):
from itertools import accumulate

n, k = list(map(int,input().split(" ")))
s = input()


def solve():
    cnt=0
    l=[]
    prev = "1"

    for i in s:
        if prev ==i:
            cnt+=1
        else:
            l.append(cnt)
            cnt = 1
        prev=i
    l.append(cnt)
    if prev=="0":
        l.append("0")

    w = 2*k+1

    if len(l)<=w:
        return n

    acc = [0]+list(accumulate(l))
    ans = 0
    for i in range(0, len(l)-w+1,2):
        ans = max(ans,acc[i+w]-acc[i])
    return ans
print(solve())

Buggy submission (verdict: Runtime Error):
from itertools import accumulate

n,k = list(map(int,input().split(" ")))
s = input()

def solve():
    cnt=0
    l=[]
    prev = "1"
    for i in s:
        if prev ==i:
            cnt += 1
        else:
            l.append(cnt)
            cnt = 1
        prev =i
    l.append(cnt)
    if prev=="0":
        l.append("0")

    w = 2 * k + 1

    if len(l) <=w:
        return n

    acc=[0]+list(accumulate(l))
    ans = 0
    for i in range(0,len(l) - w+ 1,2):
        ans = max(ans, acc[i+w] - acc[i])

    return ans
print(solve())

Buggy submission (verdict: Runtime Error):
from itertools import accumulate

n,k = list(map(int,input().split(" ")))
s = input()

def solve():
    cnt=0
    l=[]
    prev = "1"
    for i in s:
        if prev ==i:
            cnt += 1
        else:
            l.append(cnt)
            cnt = 1
        prev =i
    l.append(cnt)
    if prev=="0":
        l.append("0")

    w = 2 * k + 1

    if len(l) <=w:
        return n

    acc=[0]+list(accumulate(l))
    ans = 0
    for i in range(0,len(l) - w+ 1,2):
        ans = max(ans, acc[i+w] - acc[i])

    return ans
print(solve())

Buggy submission (verdict: Runtime Error):
from itertools import accumulate

n,k = list(map(int,input().split(" ")))
s = input()

def solve():
    cnt=0
    l=[]
    prev = "1"
    for i in s:
        if prev ==i:
            cnt += 1
        else:
            l.append(cnt)
            cnt = 1
        prev =i
    l.append(cnt)
    if prev=="0":
        l.append("0")

    w = 2 * k + 1

    if len(l) <=w:
        return n

    acc=[0]+list(accumulate(l))
    ans = 0
    for i in range(0,len(l) - w+ 1,2):
        ans = max(ans, acc[i+w] - acc[i])

    return ans
print(solve())

Buggy submission (verdict: Runtime Error):
from itertools import accumulate

N,K=list(map(int,input().split(" ")))
S=input()

def solve():
    cnt=0
    l=[]
    prev = "1"
    for i in S:
        if prev ==i:
            cnt += 1
        else:
            l.append(cnt)
            cnt = 1
        prev =i
    l.append(cnt)
    if prev=="0":
        l.append("0")
    
    w = 2 * K + 1
    
    if len(l) <=w:
        return N
    
    acc=[0]+list(accumulate(l))
    ans = 0
    for i in range(0,len(l) - w+ 1,2):
        ans = max(ans, acc[w+i] - acc[i])

    return ans
print(solve())

Buggy submission (verdict: Runtime Error):
from itertools import accumulate

N,K=list(map(int,input().split(" ")))
S=input()
def solve():
    cnt=0
    l=[]
    prev = "1"
    for i in S:
        if prev ==i:
            cnt+=1
        else:
            l.append(cnt)
            cnt = 1
        prev =i
    l.append(cnt)
    if prev=="0":
        l.append("0")
    w = 2*K+1
    if len(l) <=w:
        return N
    acc=[0]+list(accumulate(l))
    ans = 0
    for i in range(0,len(l)-w+1,2):
        ans = max(ans,acc[w+i]-acc[i])

    return ans
print(solve())