Task: | Kortit II |
Sender: | jokeri2222 |
Submission time: | 2024-11-05 13:15:28 +0200 |
Language: | Python3 (PyPy3) |
Status: | READY |
Result: | 0 |
group | verdict | score |
---|---|---|
#1 | TIME LIMIT EXCEEDED | 0 |
#2 | TIME LIMIT EXCEEDED | 0 |
#3 | TIME LIMIT EXCEEDED | 0 |
#4 | TIME LIMIT EXCEEDED | 0 |
#5 | TIME LIMIT EXCEEDED | 0 |
test | verdict | time | group | |
---|---|---|---|---|
#1 | TIME LIMIT EXCEEDED | -- | 1, 2, 3, 4, 5 | details |
#2 | TIME LIMIT EXCEEDED | -- | 2, 3, 4, 5 | details |
#3 | TIME LIMIT EXCEEDED | -- | 3, 4, 5 | details |
#4 | TIME LIMIT EXCEEDED | -- | 4, 5 | details |
#5 | TIME LIMIT EXCEEDED | -- | 5 | details |
#6 | TIME LIMIT EXCEEDED | -- | 5 | details |
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: TIME LIMIT EXCEEDED
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: TIME LIMIT EXCEEDED
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: TIME LIMIT EXCEEDED
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: TIME LIMIT EXCEEDED
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: TIME LIMIT EXCEEDED
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: TIME LIMIT EXCEEDED
input |
---|
1000 2000 1107 508 2000 1372 249 2000 588 65 2000 1739 78 ... |
correct output |
---|
750840601 678722180 744501884 159164549 868115056 ... |
user output |
---|
(empty) |