198682025-12-28 11:47:06marnorlorÚthasználati díj (75 pont)python3Hibás válasz 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()
RészfeladatÖsszpontTesztVerdiktIdőMemória
base60/75
1Elfogadva0/016ms3124 KiB
2Elfogadva0/016ms3316 KiB
3Hibás válasz0/052ms4956 KiB
4Elfogadva5/516ms3124 KiB
5Elfogadva5/516ms2952 KiB
6Hibás válasz0/516ms3124 KiB
7Elfogadva5/517ms3120 KiB
8Elfogadva5/516ms3124 KiB
9Elfogadva5/516ms3100 KiB
10Elfogadva5/517ms3120 KiB
11Elfogadva5/516ms3052 KiB
12Elfogadva5/517ms3124 KiB
13Hibás válasz0/516ms3120 KiB
14Elfogadva5/537ms5048 KiB
15Elfogadva5/537ms4980 KiB
16Elfogadva5/539ms4920 KiB
17Elfogadva5/541ms5044 KiB
18Hibás válasz0/550ms5044 KiB