CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:jokeri2222
Submission time:2024-11-05 13:15:28 +0200
Language:Python3 (PyPy3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
Test results
testverdicttimegroup
#1--1, 2, 3, 4, 5details
#2--2, 3, 4, 5details
#3--3, 4, 5details
#4--4, 5details
#5--5details
#6--5details

Code

import copy
from itertools import combinations
import math

def ith_permutation(l, i):
    n=len(l)
    fact = [1] * n
    perm = [0] * n

    for k in range(1, n):
        fact[k] = fact[k - 1] * k

    for k in range(n):
        perm[k] = i // fact[n - 1 - k]
        i = i % fact[n - 1 - k]

    for k in range(n - 1, 0, -1):
        for j in range(k - 1, -1, -1):
            if perm[j] <= perm[k]:
                perm[k] += 1

    return map(lambda x: l[x], perm)
def nCr(n, k):
    return int(math.factorial(n)/math.factorial(n-k))

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
    draws = range(n-apb)
    pairs = list(range(n))

    for draw in sorted(draws, reverse=True):
        pairs.pop(draw)
    i = _fac
    while i >= 0:
        perm = ith_permutation(pairs, i)
        A = B = 0
        for j, (pair1, pair2) in enumerate(zip(pairs, perm)):
            if pair2 < pair1:
                A += 1
            elif pair2 > pair1:
                B += 1

            if A > a or B > b or (A<=a and B<=b and j==apb-1):
                i -= math.factorial(apb - j - 1)
                break

        if A == a and B == b:
            possibles += 1

    possibles = possibles * math.factorial(n) * nCr(n, n-apb) % 1000000007
    print(possibles)

Test details

Test 1

Group: 1, 2, 3, 4, 5

Verdict:

input
54
4 4 0
3 1 3
3 2 2
4 0 4
...

correct output
0
0
0
0
0
...

user output
(empty)

Test 2

Group: 2, 3, 4, 5

Verdict:

input
284
6 1 0
5 0 2
7 1 5
7 7 5
...

correct output
0
0
35280
0
36720
...

user output
(empty)

Test 3

Group: 3, 4, 5

Verdict:

input
841
19 3 12
19 19 13
19 7 13
20 11 15
...

correct output
40291066
0
0
0
0
...

user output
(empty)

Test 4

Group: 4, 5

Verdict:

input
1000
15 12 6
7 1 6
44 4 26
6 6 5
...

correct output
0
5040
494558320
0
340694548
...

user output
(empty)

Test 5

Group: 5

Verdict:

input
1000
892 638 599
966 429 655
1353 576 1140
1403 381 910
...

correct output
0
0
0
249098285
0
...

user output
(empty)

Test 6

Group: 5

Verdict:

input
1000
2000 1107 508
2000 1372 249
2000 588 65
2000 1739 78
...

correct output
750840601
678722180
744501884
159164549
868115056
...

user output
(empty)