Kortit II

import itertools as itt
import math



test_amount = int(input())
input_list = []
solution_list = []
solutions = 0
a_wins = 0
b_wins = 0
draws = 0
cards_amount = 0
wins = 0



for _ in range(test_amount):
    solutions = 0
    
    input_list = [int(x) for x in input().split()]

    cards_amount = input_list[0]
    a_wins = input_list[1]
    b_wins = input_list[2]
    if b_wins < a_wins:
        wins = b_wins
    else:
        wins = a_wins


    draws = cards_amount - a_wins - b_wins


    if draws >= 0 and not (a_wins != 0 and b_wins == 0) and not (b_wins != 0 and a_wins == 0):
        
        combinations = itt.combinations(range(cards_amount-draws-2, 0, -1), wins-1)


        for i in combinations:
            solutions += (cards_amount-draws-1)*math.prod(i)-1
        
        solutions = (math.factorial(cards_amount)/math.factorial(draws)/math.factorial(cards_amount-draws)*solutions*math.factorial(cards_amount))%(10**9+7)
    
    else:
        solutions = 0
    
    solution_list.append(int(solutions))


for sol in solution_list:
    print(sol)

Test 1

Group: 1, 2, 3, 4, 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 37, in <module>
    combinations = itt.combinations(range(cards_amount-draws-2, 0, -1), wins-1)
ValueError: r must be non-negative

Test 2

Group: 2, 3, 4, 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 37, in <module>
    combinations = itt.combinations(range(cards_amount-draws-2, 0, -1), wins-1)
ValueError: r must be non-negative

Test 3

Group: 3, 4, 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 37, in <module>
    combinations = itt.combinations(range(cards_amount-draws-2, 0, -1), wins-1)
ValueError: r must be non-negative

Test 4

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 5

Verdict: TIME LIMIT EXCEEDED