Torni

import math
from decimal import *

def binom(n, k):
    x = math.factorial(n) // math.factorial(k) // math.factorial(n - k)
    return x

def torni(t):
    koko = 0
    for i in range(0,t+1):
        koko += ((binom(t,i))*2*((t-i)**2))
    koko = koko-t-1
    tulos = koko%((10**9)+7)
    return tulos
                           

if __name__ == '__main__':
    print(torni(6))
    print(torni(2864))
    print('')
    print(binom(200000,12518))

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