CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:hertsu
Submission time:2024-11-09 14:16:48 +0200
Language:Python3 (PyPy3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
Test results
testverdicttimegroup
#10.07 s1, 2, 3, 4, 5details
#20.07 s2, 3, 4, 5details
#30.09 s3, 4, 5details
#40.09 s4, 5details
#50.08 s5details
#60.08 s5details

Code

def yli(n, m):
    number1 = 1
    number2 = 1
    for i in range(min(m, n-m)):
        number1 *= (n-i)
        number2 *= i+1
    return number1/number2
kertomat = {}
def kertoma(n):
    if n == 0:
        return 1
    elif n in kertomat:
        return kertomat[n]
    else:
        return kertoma(n-1)*n

def calculateInputCombinations(input):
    [sticks, awins, bwins] = map(int, input.split(" "))
    draws = sticks - awins - bwins
    if draws<0:
        print(0)
        return
    if (awins==0 or bwins==0) and sticks != draws:
        print(0)
        return
    drawCombinations = yli(sticks, draws)
    n=sticks-draws
    small = min(awins, bwins)
    large = max(awins, bwins)
    hajonta = large - (small + large)/2
    if(min(awins, bwins) == 1):
        print(int(drawCombinations*kertoma(sticks)%(10**9+7)))
    else:
        print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-hajonta)*(large+hajonta)))+1))%(10**9+7)))




n = int(input())
inputs = []
for i in range(n):
    inputs.append(input())
for i in inputs:
    calculateInputCombinations(i)

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
0
0
0
0
0

Error:
Traceback (most recent call last):
  File "input/code.py", line 44, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 34, in calculateInputCombinations
    print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-hajonta)*(large+hajonta)))+1))%(10**9+7)))
ZeroDivisionError: float division by zero

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
0
0
35280
0
104400
...

Error:
Traceback (most recent call last):
  File "input/code.py", line 44, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 34, in calculateInputCombinations
    print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-hajonta)*(large+hajonta)))+1))%(10**9+7)))
ZeroDivisionError: float division by zero

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
280168798
0
0
0
0
...

Error:
Traceback (most recent call last):
  File "input/code.py", line 44, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 34, in calculateInputCombinations
    print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-hajonta)*(large+hajonta)))+1))%(10**9+7)))
ZeroDivisionError: float division by zero

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
0
5040
262622009
0
236951779
...

Error:
Traceback (most recent call last):
  File "input/code.py", line 44, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 34, in calculateInputCombinations
    print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-hajonta)*(large+hajonta)))+1))%(10**9+7)))
ZeroDivisionError: float division by zero

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
0
0
0

Error:
Traceback (most recent call last):
  File "input/code.py", line 44, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 34, in calculateInputCombinations
    print(int((drawCombinations*kertoma(sticks)*(kertoma(n)/(((small-hajonta)*(large+hajonta)))+1))%(10**9+7)))
OverflowError: int too large to convert to 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 "input/code.py", line 44, in <module>
    calculateInputCombinations(i)
  File "input/code.py", line 26, in calculateInputCombinations
    drawCombinations = yli(sticks, draws)
  File "input/code.py", line 7, in yli
    return number1/number2
OverflowError: integer division result too large for a float