198702025-12-28 12:43:05marnorlorÚthasználati díj (75 pont)python3Hibás válasz 0/7517ms3600 KiB
def szamolas(N: int, K: int, dij: list):
    varosok=0
    eredmeny=0
    for i in range(0, len(dij)):
        if dij[i]==1:
            varosok=varosok+1
    if varosok==0 or varosok==1:
        return 0
    else:
        for i in range(2, K+1):
            if i>varosok:
                break
            eredmeny=eredmeny+(varosok-(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

    a=szamolas(6, 5, [1, 0, 1, 0, 1, 0])
    print(a)
    assert a==3
    
    a=szamolas(15, 5, [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1])
    print(a)
    assert a==50
    
    a=szamolas(2, 2, [1, 1])
    print(a)
    assert a==1


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
base0/75
1Hibás válasz0/016ms3308 KiB
2Hibás válasz0/017ms3124 KiB
3Hibás válasz0/017ms3008 KiB
4Hibás válasz0/516ms3148 KiB
5Hibás válasz0/517ms3148 KiB
6Hibás válasz0/516ms3096 KiB
7Hibás válasz0/516ms3116 KiB
8Hibás válasz0/516ms3228 KiB
9Hibás válasz0/516ms3124 KiB
10Hibás válasz0/516ms3148 KiB
11Hibás válasz0/516ms3124 KiB
12Hibás válasz0/516ms3120 KiB
13Hibás válasz0/516ms3124 KiB
14Hibás válasz0/517ms3376 KiB
15Hibás válasz0/516ms3124 KiB
16Hibás válasz0/516ms3600 KiB
17Hibás válasz0/516ms3012 KiB
18Hibás válasz0/517ms3112 KiB