結果
| 問題 | No.1195 数え上げを愛したい（文字列編） |
| ユーザー |
 とりゐ
|
| 提出日時 | 2022-01-08 22:45:39 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,415 bytes |
| コンパイル時間 | 369 ms |
| コンパイル使用メモリ | 82,000 KB |
| 実行使用メモリ | 245,180 KB |
| 最終ジャッジ日時 | 2024-04-26 19:31:23 |
| 合計ジャッジ時間 | 8,849 ms |
|
ジャッジサーバーID （参考情報） |
judge5 / judge3 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | TLE | - |
| testcase_01 | -- | - |
| testcase_02 | -- | - |
| testcase_03 | -- | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
| testcase_25 | -- | - |
ソースコード
p, g = 998244353, 3
invg = pow(g, p-2, p)
W = [pow(g, (p - 1) >> i, p) for i in range(24)]
iW = [pow(invg, (p - 1) >> i, p) for i in range(24)]
def fft(k, f):
for l in range(k)[::-1]:
d = 1 << l
u = 1
for i in range(d):
for j in range(i, 1 << k, 2*d):
f[j], f[j+d] = (f[j] + f[j+d]) % p, u * (f[j] - f[j+d]) % p
u = u * W[l+1] % p
def ifft(k, f):
for l in range(k):
d = 1 << l
u = 1
for i in range(d):
for j in range(i, 1 << k, 2*d):
f[j+d] *= u
f[j], f[j+d] = (f[j] + f[j+d]) % p, (f[j] - f[j+d]) % p
u = u * iW[l+1] % p
def convolve(a, b):
n0, n1 = len(a), len(b)
k = (max(n0, n1) - 1).bit_length() + 1
n = 1 << k
a = a + [0] * (n-n0)
b = b + [0] * (n-n1)
fft(k, a), fft(k, b)
for i in range(n):
a[i] = a[i] * b[i] % p
ifft(k, a)
invn = pow(n, p - 2, p)
return [a[i] * invn % p for i in range(n0 + n1 - 1)]
s=input()
n=len(s)
mod=998244353
cnt=[0]*26
for i in s:
cnt[ord(i)-97]+=1
dp=[0]*(n+1)
dp[0]=1
A=[1]
f=[1]
finv=[1]
tmp=1
a=[1]
for i in range(1,n+1):
tmp*=i
tmp%=mod
f.append(tmp)
finv.append(pow(tmp,mod-2,mod))
for i in range(26):
b=[1]
for j in range(1,cnt[i]+1):
b.append(finv[j])
a=convolve(a, b)
#print(a)
ans=0
for i in range(1,n+1):
ans+=a[i]*f[i]
ans%=mod
print(ans)