結果

問題 No.3178 free sort
ユーザー Hyoka7
提出日時 2025-06-13 22:52:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 74 ms / 2,000 ms
コード長 696 bytes
コンパイル時間 322 ms
コンパイル使用メモリ 82,600 KB
実行使用メモリ 79,884 KB
最終ジャッジ日時 2025-06-13 22:52:11
合計ジャッジ時間 4,961 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import Counter

mod = 998244353
MAX = 2 * 10**5 + 5
factorial = [1] * MAX
inverse = [1] * MAX
for i in range(1, MAX):
    factorial[i] = factorial[i - 1] * i % mod
inverse[MAX - 1] = pow(factorial[MAX - 1], mod - 2, mod)
for i in range(MAX - 2, -1, -1):
    inverse[i] = inverse[i + 1] * (i + 1) % mod
s = input()
s_counter = Counter(s)
len_s = len(s)
all = factorial[len_s]
for val in s_counter.values():
    all = all * inverse[val] % mod
if s_counter["0"] == 0:
    print(all)
    exit()
s_counter["0"] -= 1
len_s -= 1
invalid = factorial[len_s]
for val in s_counter.values():
    invalid = invalid * inverse[val] % mod
s_counter["0"] += 1

print((all - invalid + mod) % mod)
0