cuu minh voi can gap nha

from collections import defaultdict
k=int(input())
s=input()
n=len(s)
a=defaultdict(lambda : 0)
a[0]=1
b=0
d=0
for i in range(n-1,-1,-1):
    b+=ord(s[i])-48
    if b>=k:
        d+=a[b-k]
    a[b]+=1
print(d)

n    = int(input())
d    = [0]*(10**6+1)
str  = input()
d[0] = 1

cur  = 0
res  = 0
for x in str:
    if x=="1": 
        cur += 1
    if cur >= n:
        res += d[cur-n]
    d[cur] +=1
print(res)
@hoangduy67