Kortit II

def yli(n, m):
    number1 = 1
    number2 = 1
    for i in range(min(m, n-m)):
        number1 *= (n-i)
        number2 *= i+1
    return number1/number2
kertomat = {}
def kertoma(n):
    if n == 0:
        return 1
    elif n in kertomat:
        return kertomat[n]
    else:
        return kertoma(n-1)*n

def calculateInputCombinations(input):
    [sticks, awins, bwins] = map(int, input.split(" "))
    draws = sticks - awins - bwins
    if draws<0:
        print(0)
        return
    if (awins==0 or bwins==0) and sticks != draws:
        print(0)
        return
    drawCombinations = yli(sticks, draws)
    n=sticks-draws
    small = min(awins, bwins)
    large = max(awins, bwins)
    if(min(awins, bwins) == 1):
        print(int(drawCombinations*kertoma(sticks)%(10**9+7)))
    else:
        print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-(small-2))*(large+(small - 2))))+1))%(10**9+7)))




n = int(input())
inputs = []
for i in range(n):
    inputs.append(input())
for i in inputs:
    calculateInputCombinations(i)

Test 1

Group: 1, 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 2

Group: 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 3

Group: 3, 4, 5

Verdict: WRONG ANSWER

Test 4

Group: 4, 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 43, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 33, in calculateInputCombinations
    print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-(small-2))*(large+(small - 2))))+1))%(10**9+7)))
ValueError: cannot convert float NaN to integer

Test 5

Group: 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 43, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 33, in calculateInputCombinations
    print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-(small-2))*(large+(small - 2))))+1))%(10**9+7)))
OverflowError: int too large to convert to float

Test 6

Group: 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 43, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 26, in calculateInputCombinations
    drawCombinations = yli(sticks, draws)
  File "input/code.py", line 7, in yli
    return number1/number2
OverflowError: integer division result too large for a float