結果

問題 No.2141 Enumeratest
ユーザー kawa-yo
提出日時 2022-12-04 16:51:50
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 744 bytes
コンパイル時間 123 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 128,444 KB
最終ジャッジ日時 2024-10-11 19:42:32
合計ジャッジ時間 57,954 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
num = 1000001

fact = [0]*num
inv = [0]*num
fact_inv = [0]*num

def comb_init():
    fact[0] = fact[1] = 1
    fact_inv[0] = fact_inv[1] = 1
    inv[1] = 1

    for i in range(2, num):
        fact[i] = fact[i-1] * i % mod
        inv[i] = mod - inv[mod%i] * (mod // i) % mod
        fact_inv[i] = fact_inv[i-1] * inv[i] % mod

comb_init()

def comb(n, m):
    if m > n - m:
        return comb(n, n - m)

    return fact[n] * fact_inv[m] * fact_inv[n-m] % mod

n, m = map(int, input().split(" "))


avg = m // n
res = m % n

ans = 1
cnt = 0
for i in range(n - res):
    ans *= comb(m - cnt, avg)
    ans %= mod
    cnt += avg
for i in range(n-res, n):
    ans *= comb(m - cnt, avg+1)
    ans %= mod
    cnt += avg+1

print(ans)
0