結果
問題 | No.1102 Remnants |
ユーザー | neterukun |
提出日時 | 2020-07-03 22:34:12 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 367 ms / 2,000 ms |
コード長 | 2,470 bytes |
コンパイル時間 | 330 ms |
コンパイル使用メモリ | 81,684 KB |
実行使用メモリ | 186,228 KB |
最終ジャッジ日時 | 2023-10-17 05:34:31 |
合計ジャッジ時間 | 7,634 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 138 ms
158,568 KB |
testcase_01 | AC | 137 ms
158,568 KB |
testcase_02 | AC | 137 ms
158,568 KB |
testcase_03 | AC | 137 ms
158,568 KB |
testcase_04 | AC | 136 ms
158,568 KB |
testcase_05 | AC | 285 ms
184,304 KB |
testcase_06 | AC | 168 ms
160,984 KB |
testcase_07 | AC | 367 ms
185,800 KB |
testcase_08 | AC | 146 ms
158,784 KB |
testcase_09 | AC | 151 ms
158,788 KB |
testcase_10 | AC | 146 ms
158,784 KB |
testcase_11 | AC | 150 ms
158,784 KB |
testcase_12 | AC | 154 ms
158,788 KB |
testcase_13 | AC | 151 ms
158,788 KB |
testcase_14 | AC | 222 ms
170,772 KB |
testcase_15 | AC | 205 ms
167,712 KB |
testcase_16 | AC | 318 ms
185,644 KB |
testcase_17 | AC | 329 ms
186,228 KB |
testcase_18 | AC | 332 ms
164,100 KB |
testcase_19 | AC | 211 ms
166,832 KB |
testcase_20 | AC | 161 ms
159,256 KB |
testcase_21 | AC | 255 ms
175,468 KB |
testcase_22 | AC | 305 ms
183,656 KB |
testcase_23 | AC | 268 ms
176,372 KB |
testcase_24 | AC | 230 ms
170,008 KB |
testcase_25 | AC | 334 ms
164,112 KB |
testcase_26 | AC | 155 ms
158,788 KB |
testcase_27 | AC | 194 ms
165,308 KB |
ソースコード
""" 愚直解 from collections import defaultdict def solve(ans, cnt): if cnt == k: return n = len(ans) for l in range(n): for r in range(l + 1, n + 1): if cnt == k - 1: memo[len(ans[l:r])] += 1 solve(ans[l:r], cnt + 1) n, k = map(int, input().split()) a = list(map(int, input().split())) MOD = 10 ** 9 + 7 memo = defaultdict(int) ans = [1, 2, 3, 4, 5, 6] k = 4 solve(ans, 0) print(memo) """ class Combination: """階乗とその逆元のテーブルをO(N)で事前作成し、組み合わせの計算をO(1)で行う""" def __init__(self, n, MOD): self.fact = [1] for i in range(1, n + 1): self.fact.append(self.fact[-1] * i % MOD) self.inv_fact = [0] * (n + 1) self.inv_fact[n] = pow(self.fact[n], MOD - 2, MOD) for i in reversed(range(n)): self.inv_fact[i] = self.inv_fact[i + 1] * (i + 1) % MOD self.MOD = MOD def inverse(self, k): """kの逆元を求める O(1)""" return (self.inv_fact[k] * self.fact[k - 1]) % self.MOD def factorial(self, k): """k!を求める O(1)""" return self.fact[k] def inverse_factorial(self, k): """k!の逆元を求める O(1)""" return self.inv_fact[k] def permutation(self, k, r): """kPrを求める O(1)""" if k < r: return 0 return (self.fact[k] * self.inv_fact[k - r]) % self.MOD def combination(self, k, r): """kCrを求める O(1)""" if k < r: return 0 return (self.fact[k] * self.inv_fact[k - r] * self.inv_fact[r]) % self.MOD def combination2(self, k, r): """kCrを求める O(r) kが大きいが、r <= nを満たしているときに使用 """ if k < r: return 0 res = 1 for l in range(r): res *= (k - l) res %= self.MOD return (res * self.inv_fact[r]) % self.MOD n, k = map(int, input().split()) a = list(map(int, input().split())) MOD = 10 ** 9 + 7 comb = Combination(10 ** 6, MOD) ball = k dp = [0] * (n + 10) dp[0] = 1 for i in range(n + 9): dp[i + 1] = dp[i] * (ball + i + 1) * pow(i + 1, MOD - 2, MOD) dp[i + 1] %= MOD ball = k ans = 0 for i in range(n): tmp = 1 box = n - i tmp *= dp[box - 1] box = i + 1 tmp *= dp[box - 1] ans += a[i] * tmp % MOD ans %= MOD print(ans % MOD)