提出 #40055409

ソースコード 拡げる

from itertools import combinations_with_replacement as cr
from collections import defaultdict
from math import factorial
from functools import reduce
from operator import mul

MOD = 998244353
N, M, K = map(int, input().split())
*A, = map(int, input().replace("0", "").split())

zeros = N - len(A)

ans = defaultdict(int)
for com in cr(range(1, M + 1), zeros):
  #print(com)
  ns = defaultdict(int)
  for c in com:
    ns[c] += 1
  size = factorial(zeros) // reduce(mul, [factorial(v) for v in ns.values()])
  #print(sorted(A + list(com)))
  x = sorted(A + list(com))[K-1]
  ans[x] += size
  ans[x] %= MOD
  
pat = sum([k * v % MOD for k, v in ans.items()]) % MOD
total = pow(pow(M, -1, MOD), zeros, MOD)
print(pat * total % MOD)  

#print(ans)

提出情報

提出日時
問題 E - Kth Number
ユーザ arakaki_tokyo
言語 Python (3.8.2)
得点 0
コード長 773 Byte
結果 RE
実行時間 4411 ms
メモリ 9632 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 0 / 500
結果
AC × 3
AC × 3
TLE × 36
RE × 3
セット名 テストケース
Sample 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt
All 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt, 01_random_00.txt, 01_random_01.txt, 01_random_02.txt, 01_random_03.txt, 01_random_04.txt, 01_random_05.txt, 01_random_06.txt, 01_random_07.txt, 01_random_08.txt, 01_random_09.txt, 01_random_10.txt, 01_random_11.txt, 01_random_12.txt, 01_random_13.txt, 01_random_14.txt, 01_random_15.txt, 01_random_16.txt, 01_random_17.txt, 01_random_18.txt, 01_random_19.txt, 01_random_20.txt, 01_random_21.txt, 01_random_22.txt, 01_random_23.txt, 01_random_24.txt, 01_random_25.txt, 01_random_26.txt, 01_random_27.txt, 01_random_28.txt, 01_random_29.txt, 02_k_eq_1_00.txt, 02_k_eq_1_01.txt, 03_k_eq_n_00.txt, 03_k_eq_n_01.txt, 04_small_kind_00.txt, 04_small_kind_01.txt, 04_small_kind_02.txt, 05_handmade_00.txt, 05_handmade_01.txt
ケース名 結果 実行時間 メモリ
00_sample_00.txt AC 29 ms 9404 KiB
00_sample_01.txt AC 21 ms 9404 KiB
00_sample_02.txt AC 513 ms 9404 KiB
01_random_00.txt RE 21 ms 9592 KiB
01_random_01.txt RE 25 ms 9540 KiB
01_random_02.txt RE 21 ms 9484 KiB
01_random_03.txt TLE 4410 ms 9204 KiB
01_random_04.txt TLE 4411 ms 9416 KiB
01_random_05.txt TLE 4411 ms 9316 KiB
01_random_06.txt TLE 4411 ms 9404 KiB
01_random_07.txt TLE 4410 ms 9208 KiB
01_random_08.txt TLE 4410 ms 9632 KiB
01_random_09.txt TLE 4410 ms 9508 KiB
01_random_10.txt TLE 4410 ms 9204 KiB
01_random_11.txt TLE 4410 ms 9592 KiB
01_random_12.txt TLE 4410 ms 9348 KiB
01_random_13.txt TLE 4410 ms 9260 KiB
01_random_14.txt TLE 4410 ms 9512 KiB
01_random_15.txt TLE 4410 ms 9348 KiB
01_random_16.txt TLE 4410 ms 9324 KiB
01_random_17.txt TLE 4410 ms 9332 KiB
01_random_18.txt TLE 4410 ms 9248 KiB
01_random_19.txt TLE 4410 ms 9352 KiB
01_random_20.txt TLE 4410 ms 9388 KiB
01_random_21.txt TLE 4410 ms 9212 KiB
01_random_22.txt TLE 4410 ms 9368 KiB
01_random_23.txt TLE 4410 ms 9464 KiB
01_random_24.txt TLE 4410 ms 9424 KiB
01_random_25.txt TLE 4410 ms 9356 KiB
01_random_26.txt TLE 4410 ms 9448 KiB
01_random_27.txt TLE 4410 ms 9428 KiB
01_random_28.txt TLE 4410 ms 9324 KiB
01_random_29.txt TLE 4410 ms 9340 KiB
02_k_eq_1_00.txt TLE 4410 ms 9216 KiB
02_k_eq_1_01.txt TLE 4410 ms 9300 KiB
03_k_eq_n_00.txt TLE 4410 ms 9328 KiB
03_k_eq_n_01.txt TLE 4410 ms 9200 KiB
04_small_kind_00.txt TLE 4410 ms 9312 KiB
04_small_kind_01.txt TLE 4410 ms 9324 KiB
04_small_kind_02.txt TLE 4410 ms 9396 KiB
05_handmade_00.txt TLE 4410 ms 9468 KiB
05_handmade_01.txt TLE 4410 ms 9516 KiB