Kortit II

import itertools as itt
import math



test_amount = int(input())
input_list = []
solution_list = []
a_cards = []
solution = 0
a_wins = 0
b_wins = 0
a_greater = 0
b_greater = 0
draws = 0
cards_amount = 0
rep_draws = 0



for _ in range(test_amount):
    solution = 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]

    draws = cards_amount - a_wins - b_wins

    
    a_cards = range(1, cards_amount+1-draws)
    
    permutations = itt.permutations(a_cards, len(a_cards))

    if draws >= 0 and not (a_wins != 0 and b_wins == 0) and not (b_wins != 0 and a_wins == 0):
        for b_cards in permutations:
            a_greater = 0
            b_greater = 0
            for a, b in zip(a_cards, b_cards):
                if a>b:
                    a_greater += 1
                elif b>a:
                    b_greater += 1
                elif a == b:
                    break

                if a_greater>a_wins or b_greater>b_wins:
                    break
            
            if a_greater == a_wins and b_greater == b_wins:
                solution += 1
        if draws:
            solution = (math.factorial(cards_amount)/math.factorial(draws)/math.factorial(cards_amount-draws)*solution*math.factorial(cards_amount))%(10**9+7)
        else:
            solution = (solution*math.factorial(cards_amount))%(10**9+7)
    else:
        solution = 0
    
    solution_list.append(int(solution))


for sol in solution_list:
    print(sol)

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

Test 2

Group: 2, 3, 4, 5

Verdict: ACCEPTED

Test 3

Group: 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

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