結果
問題 | No.1195 数え上げを愛したい(文字列編) |
ユーザー | とりゐ |
提出日時 | 2022-01-08 23:09:31 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,386 ms / 3,000 ms |
コード長 | 1,581 bytes |
コンパイル時間 | 160 ms |
コンパイル使用メモリ | 82,132 KB |
実行使用メモリ | 171,952 KB |
最終ジャッジ日時 | 2024-04-26 19:32:00 |
合計ジャッジ時間 | 23,408 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,382 ms
137,300 KB |
testcase_01 | AC | 1,380 ms
137,156 KB |
testcase_02 | AC | 1,386 ms
136,664 KB |
testcase_03 | AC | 483 ms
97,648 KB |
testcase_04 | AC | 552 ms
100,920 KB |
testcase_05 | AC | 1,382 ms
171,952 KB |
testcase_06 | AC | 41 ms
53,700 KB |
testcase_07 | AC | 39 ms
53,736 KB |
testcase_08 | AC | 334 ms
89,040 KB |
testcase_09 | AC | 1,358 ms
136,704 KB |
testcase_10 | AC | 692 ms
105,736 KB |
testcase_11 | AC | 1,276 ms
131,704 KB |
testcase_12 | AC | 1,259 ms
130,412 KB |
testcase_13 | AC | 1,172 ms
125,328 KB |
testcase_14 | AC | 651 ms
103,188 KB |
testcase_15 | AC | 698 ms
105,136 KB |
testcase_16 | AC | 679 ms
105,004 KB |
testcase_17 | AC | 336 ms
88,712 KB |
testcase_18 | AC | 1,275 ms
130,684 KB |
testcase_19 | AC | 1,269 ms
131,036 KB |
testcase_20 | AC | 1,177 ms
125,976 KB |
testcase_21 | AC | 1,279 ms
132,132 KB |
testcase_22 | AC | 745 ms
108,224 KB |
testcase_23 | AC | 41 ms
54,172 KB |
testcase_24 | AC | 41 ms
54,364 KB |
testcase_25 | AC | 40 ms
54,716 KB |
ソースコード
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)] import heapq 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 hq=[] 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]) heapq.heappush(hq,(cnt[i]+1,b)) #print(a) while len(hq)>1: sizea,a=heapq.heappop(hq) sizeb,b=heapq.heappop(hq) heapq.heappush(hq,(sizea+sizeb-1,convolve(a, b))) a=hq[0][1] ans=0 for i in range(1,n+1): ans+=a[i]*f[i] ans%=mod print(ans)