Kortit II

from itertools import permutations, islice, combinations
import math
def permutation(l, n):
    return next(islice(permutations(l), n, None), None)
 
 
t = int(input())
vals = []
for test in range(t):
    vals.append(map(int, input().split()))
 
for test in range(t):
    n, a, b = vals[test]
    if a + b == 1 or a + b > n:
        print(0)
        continue
    possibles = 0
    apb = a+b
    _fac = math.factorial(apb)-1
    for draws in combinations(range(n), n - a - b):
        pairs = list(range(n))
 
        for draw in draws[::-1]:
            pairs.pop(draw)
        i=_fac
        while i>0:
            perm = permutation(pairs, i)
            A = B = 0
            for j, (pair1, pair2) in enumerate(zip(pairs, perm)):
                if pair2 - pair1 < 0:
                    A += 1
                elif pair2 - pair1 > 0:
                    B += 1
 
                if A>a or B>b:
                    i-=math.factorial(apb-j)-1
                    break
 
            if A == a and B == b:
                possibles += 1
 
            i-=1
 
    possibles = possibles * math.factorial(n) % 1000000007
    print(possibles)

Test 1

Group: 1, 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 2

Group: 2, 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 4, 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 27, in <module>
    perm = permutation(pairs, i)
  File "input/code.py", line 4, in permutation
    return next(islice(permutations(l), n, None), None)
ValueError: Indicies for islice() must be None or non-negative integers

Test 5

Group: 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 27, in <module>
    perm = permutation(pairs, i)
  File "input/code.py", line 4, in permutation
    return next(islice(permutations(l), n, None), None)
ValueError: Indicies for islice() must be None or non-negative integers

Test 6

Group: 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 27, in <module>
    perm = permutation(pairs, i)
  File "input/code.py", line 4, in permutation
    return next(islice(permutations(l), n, None), None)
ValueError: Indicies for islice() must be None or non-negative integers