CSES - Aalto Competitive Programming 2025 - wk11 - Mon

Submission details
Task:	Skittles
Sender:	mooc.fi~1130261
Submission time:	2025-11-17 16:59:56 +0200
Language:	Python3 (PyPy3)
Status:	READY
Result:	ACCEPTED

Test results
test	verdict	time
#1	ACCEPTED	0.04 s	details
#2	ACCEPTED	0.04 s	details
#3	ACCEPTED	0.04 s	details
#4	ACCEPTED	0.04 s	details
#5	ACCEPTED	0.04 s	details
#6	ACCEPTED	0.04 s	details
#7	ACCEPTED	0.04 s	details
#8	ACCEPTED	0.04 s	details
#9	ACCEPTED	0.04 s	details
#10	ACCEPTED	0.04 s	details
#11	ACCEPTED	0.04 s	details
#12	ACCEPTED	0.05 s	details
#13	ACCEPTED	0.05 s	details
#14	ACCEPTED	0.06 s	details
#15	ACCEPTED	0.06 s	details
#16	ACCEPTED	0.05 s	details
#17	ACCEPTED	0.06 s	details
#18	ACCEPTED	0.06 s	details
#19	ACCEPTED	0.06 s	details
#20	ACCEPTED	0.06 s	details
#21	ACCEPTED	0.06 s	details
#22	ACCEPTED	0.06 s	details
#23	ACCEPTED	0.05 s	details
#24	ACCEPTED	0.06 s	details
#25	ACCEPTED	0.05 s	details
#26	ACCEPTED	0.06 s	details
#27	ACCEPTED	0.06 s	details
#28	ACCEPTED	0.06 s	details
#29	ACCEPTED	0.06 s	details
#30	ACCEPTED	0.06 s	details
#31	ACCEPTED	0.06 s	details
#32	ACCEPTED	0.05 s	details
#33	ACCEPTED	0.06 s	details
#34	ACCEPTED	0.05 s	details
#35	ACCEPTED	0.05 s	details
#36	ACCEPTED	0.06 s	details
#37	ACCEPTED	0.06 s	details
#38	ACCEPTED	0.06 s	details
#39	ACCEPTED	0.06 s	details
#40	ACCEPTED	0.07 s	details
#41	ACCEPTED	0.05 s	details
#42	ACCEPTED	0.06 s	details
#43	ACCEPTED	0.06 s	details
#44	ACCEPTED	0.05 s	details
#45	ACCEPTED	0.06 s	details
#46	ACCEPTED	0.05 s	details
#47	ACCEPTED	0.06 s	details
#48	ACCEPTED	0.05 s	details
#49	ACCEPTED	0.06 s	details
#50	ACCEPTED	0.05 s	details
#51	ACCEPTED	0.06 s	details
#52	ACCEPTED	0.05 s	details
#53	ACCEPTED	0.06 s	details
#54	ACCEPTED	0.06 s	details
#55	ACCEPTED	0.06 s	details
#56	ACCEPTED	0.06 s	details
#57	ACCEPTED	0.06 s	details
#58	ACCEPTED	0.06 s	details
#59	ACCEPTED	0.06 s	details
#60	ACCEPTED	0.05 s	details
#61	ACCEPTED	0.06 s	details
#62	ACCEPTED	0.06 s	details
#63	ACCEPTED	0.06 s	details

Code

M = 998244353
n, a, b = map(int, input().split())
def inv(x):
    return pow(x, M - 2, M)
fact = [1] * (n + b + 1)
for i in range(1, n + b + 1):
    fact[i] = (fact[i - 1] * i) % M
invfact = [1] * (n + b + 1)
invfact[n + b] = inv(fact[n + b])
for i in range(n + b - 1, -1, -1):
    invfact[i] = (invfact[i + 1] * (i + 1)) % M
max_k = a + b - 1
ans = 0
def comb(n, k):
    return (fact[n] * invfact[k] * invfact[n - k]) % M
for i in range(a, b + 1):
    ans += comb(n + i - 1, i)
    ans %= M
print(ans)

Test details

Test 1

Verdict: ACCEPTED

input
`1 1 10` view save

correct output
`10` view save

user output
`10` view save

Test 2

Verdict: ACCEPTED

input
`2 3 3` view save

correct output
`4` view save

user output
`4` view save

Test 3

Verdict: ACCEPTED

input
`2 4 5` view save

correct output
`11` view save

user output
`11` view save

Test 4

Verdict: ACCEPTED

input
`3 4 5` view save

correct output
`36` view save

user output
`36` view save

Test 5

Verdict: ACCEPTED

input
`3 2 2` view save

correct output
`6` view save

user output
`6` view save

Test 6

Verdict: ACCEPTED

input
`5 5 5` view save

correct output
`126` view save

user output
`126` view save

Test 7

Verdict: ACCEPTED

input
`10 10 10` view save

correct output
`92378` view save

user output
`92378` view save

Test 8

Verdict: ACCEPTED

input
`10 1 10` view save

correct output
`184755` view save

user output
`184755` view save

Test 9

Verdict: ACCEPTED

input
`100 100 100` view save

correct output
`866655715` view save

user output
`866655715` view save

Test 10

Verdict: ACCEPTED

input
`10 100 100` view save

correct output
`918123961` view save

user output
`918123961` view save

Test 11

Verdict: ACCEPTED

input
`1000 10 10` view save

correct output
`84965120` view save

user output
`84965120` view save

Test 12

Verdict: ACCEPTED

input
`10000 10000 10000` view save

correct output
`658363258` view save

user output
`658363258` view save

Test 13

Verdict: ACCEPTED

input
`100000 100000 100000` view save

correct output
`966937032` view save

user output
`966937032` view save

Test 14

Verdict: ACCEPTED

input
`54883 59286 71521` view save

correct output
`908996595` view save

user output
`908996595` view save

Test 15

Verdict: ACCEPTED

input
`41703 72034 99721` view save

correct output
`139159644` view save

user output
`139159644` view save

Test 16

Verdict: ACCEPTED

input
`43601 2593 18509` view save

correct output
`804369499` view save

user output
`804369499` view save

Test 17

Verdict: ACCEPTED

input
`55081 7073 70816` view save

correct output
`806880502` view save

user output
`806880502` view save

Test 18

Verdict: ACCEPTED

input
`96705 54725 90064` view save

correct output
`521461113` view save

user output
`521461113` view save

Test 19

Verdict: ACCEPTED

input
`22200 5519 87075` view save

correct output
`866679044` view save

user output
`866679044` view save

Test 20

Verdict: ACCEPTED

input
`89288 33199 94750` view save

correct output
`812654801` view save

user output
`812654801` view save

Test 21

Verdict: ACCEPTED

input
`7631 22735 77994` view save

correct output
`30255908` view save

user output
`30255908` view save

Test 22

Verdict: ACCEPTED

input
`87345 1112 96856` view save

correct output
`944788802` view save

user output
`944788802` view save

Test 23

Verdict: ACCEPTED

input
`1038 36447 50189` view save

correct output
`654272260` view save

user output
`654272260` view save

Test 24

Verdict: ACCEPTED

input
`41703 72034 99721` view save

correct output
`139159644` view save

user output
`139159644` view save

Test 25

Verdict: ACCEPTED

input
`18028 1948 6831` view save

correct output
`631722506` view save

user output
`631722506` view save

Test 26

Verdict: ACCEPTED

input
`4873 28912 79929` view save

correct output
`106074189` view save

user output
`106074189` view save

Test 27

Verdict: ACCEPTED

input
`28606 2900 95813` view save

correct output
`538554734` view save

user output
`538554734` view save

Test 28

Verdict: ACCEPTED

input
`25093 4610 68059` view save

correct output
`799962519` view save

user output
`799962519` view save

Test 29

Verdict: ACCEPTED

input
`67575 4472 32712` view save

correct output
`857647130` view save

user output
`857647130` view save

Test 30

Verdict: ACCEPTED

input
`82449 18136 86245` view save

correct output
`420446937` view save

user output
`420446937` view save

Test 31

Verdict: ACCEPTED

input
`18558 38661 75914` view save

correct output
`924425479` view save

user output
`924425479` view save

Test 32

Verdict: ACCEPTED

input
`35543 34086 47960` view save

correct output
`210554536` view save

user output
`210554536` view save

Test 33

Verdict: ACCEPTED

input
`20101 32903 99080` view save

correct output
`88647511` view save

user output
`88647511` view save

Test 34

Verdict: ACCEPTED

input
`18028 1948 6831` view save

correct output
`631722506` view save

user output
`631722506` view save

Test 35

Verdict: ACCEPTED

input
`61218 12076 16908` view save

correct output
`820171450` view save

user output
`820171450` view save

Test 36

Verdict: ACCEPTED

input
`78139 13111 69284` view save

correct output
`904117059` view save

user output
`904117059` view save

Test 37

Verdict: ACCEPTED

input
`45455 2862 43952` view save

correct output
`430742811` view save

user output
`430742811` view save

Test 38

Verdict: ACCEPTED

input
`84538 61888 77158` view save

correct output
`294818599` view save

user output
`294818599` view save

Test 39

Verdict: ACCEPTED

input
`26937 59685 80971` view save

correct output
`680140546` view save

user output
`680140546` view save

Test 40

Verdict: ACCEPTED

input
`30632 9727 88573` view save

correct output
`877318826` view save

user output
`877318826` view save

Test 41

Verdict: ACCEPTED

input
`55737 66074 71002` view save

correct output
`123569267` view save

user output
`123569267` view save

Test 42

Verdict: ACCEPTED

input
`59351 587 51636` view save

correct output
`122513346` view save

user output
`122513346` view save

Test 43

Verdict: ACCEPTED

input
`8954 7866 51810` view save

correct output
`961929284` view save

user output
`961929284` view save

Test 44

Verdict: ACCEPTED

input
`61218 12076 16908` view save

correct output
`820171450` view save

user output
`820171450` view save

Test 45

Verdict: ACCEPTED

input
`9556 43003 92502` view save

correct output
`424034313` view save

user output
`424034313` view save

Test 46

Verdict: ACCEPTED

input
`20433 945 23474` view save

correct output
`794687216` view save

user output
`794687216` view save

Test 47

Verdict: ACCEPTED

input
`9941 45344 78340` view save

correct output
`875489658` view save

user output
`875489658` view save

Test 48

Verdict: ACCEPTED

input
`82532 68582 70655` view save

correct output
`238007640` view save

user output
`238007640` view save

Test 49

Verdict: ACCEPTED

input
`52729 37586 66259` view save

correct output
`93243919` view save

user output
`93243919` view save

Test 50

Verdict: ACCEPTED

input
`10475 17392 24871` view save

correct output
`977605482` view save

user output
`977605482` view save

Test 51

Verdict: ACCEPTED

input
`76232 88012 95143` view save

correct output
`871309467` view save

user output
`871309467` view save

Test 52

Verdict: ACCEPTED

input
`77184 58424 64500` view save

correct output
`693972156` view save

user output
`693972156` view save

Test 53

Verdict: ACCEPTED

input
`96888 47083 55295` view save

correct output
`406195062` view save

user output
`406195062` view save

Test 54

Verdict: ACCEPTED

input
`9556 43003 92502` view save

correct output
`424034313` view save

user output
`424034313` view save

Test 55

Verdict: ACCEPTED

input
`83891 47894 87748` view save

correct output
`686766192` view save

user output
`686766192` view save

Test 56

Verdict: ACCEPTED

input
`85080 21779 92819` view save

correct output
`114946782` view save

user output
`114946782` view save

Test 57

Verdict: ACCEPTED

input
`88235 6779 56603` view save

correct output
`70147249` view save

user output
`70147249` view save

Test 58

Verdict: ACCEPTED

input
`88858 66896 98375` view save

correct output
`601764178` view save

user output
`601764178` view save

Test 59

Verdict: ACCEPTED

input
`50586 8901 76084` view save

correct output
`881324074` view save

user output
`881324074` view save

Test 60

Verdict: ACCEPTED

input
`87594 26882 34225` view save

correct output
`42681251` view save

user output
`42681251` view save

Test 61

Verdict: ACCEPTED

input
`73130 30310 55103` view save

correct output
`843664756` view save

user output
`843664756` view save

Test 62

Verdict: ACCEPTED

input
`63584 31874 92876` view save

correct output
`437812641` view save

user output
`437812641` view save

Test 63

Verdict: ACCEPTED

input
`88714 2531 74051` view save

correct output
`603847190` view save

user output
`603847190` view save

Aalto Competitive Programming 2025 - wk11 - Mon

Skittles

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Test 25

Test 26

Test 27

Test 28

Test 29

Test 30

Test 31

Test 32

Test 33

Test 34

Test 35

Test 36

Test 37

Test 38

Test 39

Test 40

Test 41

Test 42

Test 43

Test 44

Test 45

Test 46

Test 47

Test 48

Test 49

Test 50

Test 51

Test 52

Test 53

Test 54

Test 55

Test 56

Test 57

Test 58

Test 59

Test 60

Test 61

Test 62

Test 63

Tasks

Submissions (mooc.fi~1130261)