198682025-12-28 11:47:06marnorlorÚthasználati díj (75 pont)python3Wrong answer 60/7552ms5048 KiB
def szamolas(N: int, K: int, dij: list):
    fizetos=[]
    eredmeny=0
    for i in range(0, len(dij)):
        if dij[i]==1:
            fizetos.append(i)
    if len(fizetos)==0 or len(fizetos)==1:
        return 0
    else:
        for i in range(2, K+1):
            eredmeny=eredmeny+(len(fizetos)-(i-1))
    return eredmeny

def test():
    a=szamolas(10, 3, [0, 1, 1, 0, 0, 0, 1, 1, 0, 1])
    assert a==7

    a=szamolas(5, 5, [0, 1, 0, 0, 0])
    assert a==0

def futtatas():
    N, K=map(int, input().split())
    dij=list(map(int, input().split()))
    a=szamolas(N, K, dij)
    print(a)

#test()
futtatas()
SubtaskSumTestVerdictTimeMemory
base60/75
1Accepted0/016ms3124 KiB
2Accepted0/016ms3316 KiB
3Wrong answer0/052ms4956 KiB
4Accepted5/516ms3124 KiB
5Accepted5/516ms2952 KiB
6Wrong answer0/516ms3124 KiB
7Accepted5/517ms3120 KiB
8Accepted5/516ms3124 KiB
9Accepted5/516ms3100 KiB
10Accepted5/517ms3120 KiB
11Accepted5/516ms3052 KiB
12Accepted5/517ms3124 KiB
13Wrong answer0/516ms3120 KiB
14Accepted5/537ms5048 KiB
15Accepted5/537ms4980 KiB
16Accepted5/539ms4920 KiB
17Accepted5/541ms5044 KiB
18Wrong answer0/550ms5044 KiB