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):
        koko += ((binom(t,i))*2*((t-i)**2))
    koko = koko-t-1
    koko = koko%((10**9)+7)
    return koko
                           

if __name__ == '__main__':
    for i in range(1000,2000):
        print(i, torni(i))
    print(torni(6))
    print(torni(2865))
    print(binom(2000,258))

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