CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:noitamonni
Submission time:2024-11-04 12:50:38 +0200
Language:Python3 (CPython3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
Test results
testverdicttimegroup
#10.02 s1, 2, 3, 4, 5details
#20.22 s2, 3, 4, 5details
#3--3, 4, 5details
#4--4, 5details
#50.02 s5details
#60.02 s5details

Code

from collections import deque
from math import factorial
import sys

t = sys.stdin.readline()

for line in sys.stdin.readlines():
    n, a, b = [int(x) for x in line.split()]
    if a + b > n: 
        print(0, end='')
        continue
    elif (a == 0 and b != 0) or (b == 0 and a != 0):
        print(0, end='')
        continue

    ties = n - a - b

    tot_cards2 = [x for x in range(1, n-ties+1)]
    valid_games = 0

    tot_runs = 0

    pos_ties = int(factorial(n)/factorial(n-ties))

    def search(c1, cards2, p1, p2):

        if c1 == 0:
            if p1 == a and p2 == b:
                global valid_games
                valid_games += 1
            
        else:
            for i in range(len(cards2)):
                
                cs2 = cards2[:]
                if len(cards2) == 0:
                    break

                if cs2[i] == c1:
                    continue
                else:
                    c2 = cs2.pop(i)
                    if c1 > c2:
                        search(c1-1, cs2, p1+1, p2)
                    elif c1 < c2:
                        search(c1-1, cs2, p1, p2+1)

    if a == 0 and b == 0:
        print(pos_ties, end='')
        continue

    if a == 1 or b == 1:
        valid_games = 1
    else:
        search(n-ties, tot_cards2, 0, 0)

    total_amount = (valid_games*factorial((n-ties)))*(factorial(n)/factorial(n-ties))**2
    print(int(total_amount % ((10**9)+7)), end='')

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
000002016806000000096096000000...

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
003528003672001159201680600000...

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
000

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 23, in <module>
    pos_ties = int(factorial(n)/factorial(n-ties))
OverflowError: integer division result too large for a float

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)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 23, in <module>
    pos_ties = int(factorial(n)/factorial(n-ties))
OverflowError: integer division result too large for a float