Torni

#n = int(input('Anna luku'))

def facto(n):
    prod = 1
    for k in range(2,n+1):
        prod = prod * k % ((10**9)+7)
    return prod

def binom(n, k):
    return (facto(n)/(facto(k)*(facto(n-k))))

def torni(t):
    koko = 0
    for i in range(0,t+1):
        x = ((binom(t,i))*2*((t-i)**2))
        koko = koko + x
    koko = koko-t-1
    return koko
                           

if __name__ == '__main__':
    print(torni(6))
    print(binom(2000,258))
    print(torni(2864))

Test 1

Group: 1, 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 2

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED