結果

問題 No.1102 Remnants
ユーザー Shinya FujitaShinya Fujita
提出日時 2024-10-04 00:27:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 275 ms / 2,000 ms
コード長 724 bytes
コンパイル時間 212 ms
コンパイル使用メモリ 82,188 KB
実行使用メモリ 134,732 KB
最終ジャッジ日時 2024-10-04 00:27:56
合計ジャッジ時間 5,431 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
53,380 KB
testcase_01 AC 38 ms
53,616 KB
testcase_02 AC 38 ms
52,536 KB
testcase_03 AC 36 ms
53,004 KB
testcase_04 AC 33 ms
53,084 KB
testcase_05 AC 231 ms
128,016 KB
testcase_06 AC 76 ms
76,916 KB
testcase_07 AC 275 ms
134,732 KB
testcase_08 AC 57 ms
67,536 KB
testcase_09 AC 57 ms
68,656 KB
testcase_10 AC 50 ms
66,492 KB
testcase_11 AC 55 ms
68,160 KB
testcase_12 AC 59 ms
69,888 KB
testcase_13 AC 57 ms
69,080 KB
testcase_14 AC 137 ms
100,536 KB
testcase_15 AC 139 ms
92,468 KB
testcase_16 AC 230 ms
116,616 KB
testcase_17 AC 232 ms
118,564 KB
testcase_18 AC 240 ms
109,068 KB
testcase_19 AC 118 ms
93,476 KB
testcase_20 AC 67 ms
74,756 KB
testcase_21 AC 166 ms
108,468 KB
testcase_22 AC 217 ms
111,396 KB
testcase_23 AC 167 ms
102,980 KB
testcase_24 AC 141 ms
100,296 KB
testcase_25 AC 243 ms
109,100 KB
testcase_26 AC 58 ms
71,392 KB
testcase_27 AC 103 ms
88,044 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

N, K = map(int, input().split())
MOD = 10**9 + 7
A = list(map(int, input().split()))
S = [0]
for a in A:
    S.append((S[-1] + a) % MOD)


B = [0] * (N+1)
B[1] = 1
b1 = b2 = 1
for i in range(2, N+1):
    b1 *= K+i-1
    b1 %= MOD
    b2 *= i-1
    b2 %= MOD
    B[i] = b1 * pow(b2, MOD-2, MOD) % MOD

C = [(B[i]-B[i-1])%MOD for i in range(N+1)]
SC = [0]
for c in C[1:]:
    SC.append((SC[-1] + c) % MOD)

SpC = [0] * (N+1)
for i in range(1, N+1):
    SpC[i] = S[i] * C[N-i+1] % MOD

SSpC = [0] * (N+1)
SSpC[N] = SpC[N]
for i in range(N-1, 0, -1):
    SSpC[i] = (SpC[i] + SSpC[i+1]) % MOD


ans = 0
for i in range(1, N+1):
    ans += C[i] * SSpC[i] % MOD
    ans -= S[i-1] * C[i] * SC[N-i+1] % MOD
    ans %= MOD

print(ans)
0