結果
問題 | No.1387 Mitarushi's Remodeling |
ユーザー | Mitarushi |
提出日時 | 2021-01-20 23:18:22 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 852 ms / 2,000 ms |
コード長 | 1,870 bytes |
コンパイル時間 | 287 ms |
コンパイル使用メモリ | 87,072 KB |
実行使用メモリ | 211,084 KB |
最終ジャッジ日時 | 2023-09-12 10:35:16 |
合計ジャッジ時間 | 51,676 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 110 ms
100,144 KB |
testcase_01 | AC | 111 ms
99,980 KB |
testcase_02 | AC | 109 ms
100,140 KB |
testcase_03 | AC | 111 ms
99,876 KB |
testcase_04 | AC | 110 ms
100,144 KB |
testcase_05 | AC | 109 ms
99,844 KB |
testcase_06 | AC | 111 ms
99,984 KB |
testcase_07 | AC | 111 ms
99,868 KB |
testcase_08 | AC | 149 ms
102,184 KB |
testcase_09 | AC | 145 ms
102,164 KB |
testcase_10 | AC | 149 ms
102,232 KB |
testcase_11 | AC | 148 ms
101,984 KB |
testcase_12 | AC | 147 ms
102,260 KB |
testcase_13 | AC | 150 ms
102,248 KB |
testcase_14 | AC | 852 ms
206,932 KB |
testcase_15 | AC | 814 ms
208,120 KB |
testcase_16 | AC | 793 ms
206,964 KB |
testcase_17 | AC | 789 ms
206,888 KB |
testcase_18 | AC | 795 ms
207,664 KB |
testcase_19 | AC | 806 ms
206,948 KB |
testcase_20 | AC | 795 ms
206,984 KB |
testcase_21 | AC | 792 ms
206,948 KB |
testcase_22 | AC | 804 ms
207,092 KB |
testcase_23 | AC | 809 ms
206,964 KB |
testcase_24 | AC | 806 ms
208,700 KB |
testcase_25 | AC | 790 ms
207,812 KB |
testcase_26 | AC | 792 ms
206,724 KB |
testcase_27 | AC | 796 ms
207,092 KB |
testcase_28 | AC | 795 ms
206,724 KB |
testcase_29 | AC | 797 ms
206,940 KB |
testcase_30 | AC | 805 ms
207,136 KB |
testcase_31 | AC | 814 ms
208,032 KB |
testcase_32 | AC | 814 ms
206,872 KB |
testcase_33 | AC | 800 ms
207,268 KB |
testcase_34 | AC | 799 ms
206,872 KB |
testcase_35 | AC | 793 ms
206,920 KB |
testcase_36 | AC | 798 ms
207,152 KB |
testcase_37 | AC | 800 ms
208,188 KB |
testcase_38 | AC | 795 ms
207,064 KB |
testcase_39 | AC | 798 ms
206,916 KB |
testcase_40 | AC | 807 ms
206,980 KB |
testcase_41 | AC | 809 ms
206,700 KB |
testcase_42 | AC | 804 ms
208,296 KB |
testcase_43 | AC | 787 ms
207,760 KB |
testcase_44 | AC | 794 ms
207,168 KB |
testcase_45 | AC | 785 ms
206,868 KB |
testcase_46 | AC | 786 ms
206,916 KB |
testcase_47 | AC | 802 ms
207,184 KB |
testcase_48 | AC | 785 ms
206,964 KB |
testcase_49 | AC | 805 ms
207,200 KB |
testcase_50 | AC | 797 ms
207,136 KB |
testcase_51 | AC | 788 ms
206,812 KB |
testcase_52 | AC | 791 ms
207,068 KB |
testcase_53 | AC | 801 ms
207,080 KB |
testcase_54 | AC | 793 ms
207,580 KB |
testcase_55 | AC | 785 ms
206,952 KB |
testcase_56 | AC | 802 ms
206,852 KB |
testcase_57 | AC | 792 ms
207,532 KB |
testcase_58 | AC | 812 ms
206,976 KB |
testcase_59 | AC | 819 ms
207,404 KB |
testcase_60 | AC | 805 ms
207,056 KB |
testcase_61 | AC | 793 ms
206,692 KB |
testcase_62 | AC | 791 ms
207,036 KB |
testcase_63 | AC | 797 ms
208,088 KB |
testcase_64 | AC | 809 ms
208,516 KB |
testcase_65 | AC | 803 ms
208,216 KB |
testcase_66 | AC | 809 ms
208,456 KB |
testcase_67 | AC | 822 ms
207,860 KB |
testcase_68 | AC | 813 ms
208,048 KB |
testcase_69 | AC | 803 ms
211,084 KB |
ソースコード
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)