Kortit II

import itertools
from sys import stdin, stdout

def factorial(x):
  result = 1
  for i in range(1, x + 1):
    result *= i
  return result

def solve(n, a, b):
  c = list(itertools.permutations(range(1, n + 1)))

  sum = 0

  for i in range(len(c)):
    asum = 0
    bsum = 0
    for k in range(n):
      if c[i][k] > k + 1:
        asum += 1
      elif c[i][k] < k + 1:
        bsum += 1
    if asum == a and bsum == b:
      sum += 1

  return sum * factorial(n)

t = int(stdin.readline())

for i in range(t):
  nab = [int(x) for x in stdin.readline().split()]
  print(solve(nab[0], nab[1], nab[2]))

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

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: TIME LIMIT EXCEEDED

Test 5

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 5

Verdict: TIME LIMIT EXCEEDED