結果
問題 | No.1387 Mitarushi's Remodeling |
ユーザー | Mitarushi |
提出日時 | 2021-01-20 22:56:41 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 833 ms / 2,000 ms |
コード長 | 1,909 bytes |
コンパイル時間 | 256 ms |
コンパイル使用メモリ | 87,120 KB |
実行使用メモリ | 210,272 KB |
最終ジャッジ日時 | 2023-09-12 10:38:29 |
合計ジャッジ時間 | 52,287 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 111 ms
100,180 KB |
testcase_01 | AC | 111 ms
99,828 KB |
testcase_02 | AC | 112 ms
100,076 KB |
testcase_03 | AC | 112 ms
100,180 KB |
testcase_04 | AC | 112 ms
100,248 KB |
testcase_05 | AC | 108 ms
100,104 KB |
testcase_06 | AC | 109 ms
100,088 KB |
testcase_07 | AC | 109 ms
99,872 KB |
testcase_08 | AC | 148 ms
102,392 KB |
testcase_09 | AC | 145 ms
102,144 KB |
testcase_10 | AC | 151 ms
102,132 KB |
testcase_11 | AC | 152 ms
102,256 KB |
testcase_12 | AC | 149 ms
102,268 KB |
testcase_13 | AC | 151 ms
102,180 KB |
testcase_14 | AC | 809 ms
206,864 KB |
testcase_15 | AC | 811 ms
206,396 KB |
testcase_16 | AC | 813 ms
206,720 KB |
testcase_17 | AC | 804 ms
206,828 KB |
testcase_18 | AC | 818 ms
206,840 KB |
testcase_19 | AC | 801 ms
206,432 KB |
testcase_20 | AC | 804 ms
206,884 KB |
testcase_21 | AC | 813 ms
207,252 KB |
testcase_22 | AC | 810 ms
206,312 KB |
testcase_23 | AC | 811 ms
206,916 KB |
testcase_24 | AC | 818 ms
206,480 KB |
testcase_25 | AC | 833 ms
207,044 KB |
testcase_26 | AC | 813 ms
206,984 KB |
testcase_27 | AC | 818 ms
206,672 KB |
testcase_28 | AC | 809 ms
206,720 KB |
testcase_29 | AC | 821 ms
206,624 KB |
testcase_30 | AC | 806 ms
206,584 KB |
testcase_31 | AC | 816 ms
206,704 KB |
testcase_32 | AC | 813 ms
206,520 KB |
testcase_33 | AC | 819 ms
206,836 KB |
testcase_34 | AC | 819 ms
206,764 KB |
testcase_35 | AC | 831 ms
206,768 KB |
testcase_36 | AC | 812 ms
207,072 KB |
testcase_37 | AC | 833 ms
206,728 KB |
testcase_38 | AC | 826 ms
207,216 KB |
testcase_39 | AC | 816 ms
206,588 KB |
testcase_40 | AC | 823 ms
206,948 KB |
testcase_41 | AC | 809 ms
206,184 KB |
testcase_42 | AC | 816 ms
206,728 KB |
testcase_43 | AC | 800 ms
206,716 KB |
testcase_44 | AC | 795 ms
206,792 KB |
testcase_45 | AC | 808 ms
207,076 KB |
testcase_46 | AC | 820 ms
207,092 KB |
testcase_47 | AC | 821 ms
207,172 KB |
testcase_48 | AC | 815 ms
207,164 KB |
testcase_49 | AC | 808 ms
206,680 KB |
testcase_50 | AC | 813 ms
206,356 KB |
testcase_51 | AC | 823 ms
206,688 KB |
testcase_52 | AC | 827 ms
206,696 KB |
testcase_53 | AC | 813 ms
206,088 KB |
testcase_54 | AC | 805 ms
206,532 KB |
testcase_55 | AC | 820 ms
207,136 KB |
testcase_56 | AC | 819 ms
206,984 KB |
testcase_57 | AC | 817 ms
206,588 KB |
testcase_58 | AC | 806 ms
206,876 KB |
testcase_59 | AC | 813 ms
206,600 KB |
testcase_60 | AC | 811 ms
207,176 KB |
testcase_61 | AC | 813 ms
206,808 KB |
testcase_62 | AC | 815 ms
207,244 KB |
testcase_63 | AC | 816 ms
207,124 KB |
testcase_64 | AC | 815 ms
207,344 KB |
testcase_65 | AC | 820 ms
207,844 KB |
testcase_66 | AC | 819 ms
207,592 KB |
testcase_67 | AC | 813 ms
207,120 KB |
testcase_68 | AC | 810 ms
207,472 KB |
testcase_69 | AC | 812 ms
210,272 KB |
ソースコード
import sys input = sys.stdin.readline mod = 998244353 max_kai = 10**6+10 kai = [1] * max_kai for i in range(1, max_kai): kai[i] = kai[i-1] * i % mod kkai = [0] * max_kai kkai[-1] = pow(kai[-1], mod - 2, mod) for i in reversed(range(1, max_kai)): kkai[i - 1] = kkai[i] * i % mod kkai_sum = [0]*(max_kai+1) for i in range(max_kai): kkai_sum[i+1] = (kkai_sum[i] + kkai[i]) % mod def cutsum(a, b): return kkai_sum[a] - kkai_sum[max(0, b)] def fft_inplace(a, w): n = len(a) m = n t = 1 while m >= 2: mh = m >> 1 for i in range(0, n, m): for s in range(mh): j, k = i+s, i+mh+s a[j], a[k] = (a[j]+a[k]) % mod, (a[j]-a[k])*w[s*t] % mod m = mh t *= 2 def ifft_inplace(a, w): n = len(a) m = 2 t = -(n >> 1) while m <= n: mh = m >> 1 for i in range(0, n, m): for s in range(mh): j, k = i+s, i+mh+s a[k] *= w[s*t] a[j], a[k] = (a[j]+a[k]) % mod, (a[j]-a[k]) % mod m <<= 1 t //= 2 n_inv = pow(n, mod-2, mod) for i in range(n): a[i] = a[i] * n_inv % mod def convolution(a, b): n2 = max(len(a) + len(b), 1) n = 1 << (n2-1).bit_length() a = a + [0] * (n-len(a)) b = b + [0] * (n-len(b)) w_root = pow(3, (mod-1)//n, mod) w = [1] * n for i in range(1, n): w[i] = w[i-1] * w_root % mod fft_inplace(a, w) fft_inplace(b, w) for i, j in enumerate(b): a[i] = a[i] * j % mod ifft_inplace(a, w) return a[:n2] n, k = map(int, input().split()) m = [int(i)+1 for i in input().split()] b = convolution(m, m[::-1]) ans = 0 for i in range(n): ans += (kai[n-1] * cutsum(n-1, n-k-1) - kai[i]*cutsum(i, i-k)) % mod * b[i] ans %= mod for i in m: ans *= i-1 ans %= mod ans *= pow(4, mod-2, mod) ans %= mod print(ans)