結果
問題 | No.1555 Constructed Balancing Sequence |
ユーザー |
![]() |
提出日時 | 2021-05-30 03:04:17 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 145 ms / 2,000 ms |
コード長 | 509 bytes |
コンパイル時間 | 332 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 21,516 KB |
最終ジャッジ日時 | 2024-06-22 19:35:29 |
合計ジャッジ時間 | 5,582 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 42 |
ソースコード
MOD = 998_244_353 def main(N, K, A): S = [0] * (N+1) for i in range(N): S[i+1] = S[i] + A[i] if any(a > b for a, b in zip(A[1:], S[1:])): return 0 v = A[-1] - 1 ans = 1 for n in range(N - 2, -1, -1): s, a = S[n], A[n] if v < s + a - 1: pass else: ans = ans * (A[n] + K + 1) % MOD v = max(A[n] - 1, v // 2) return ans N, K = map(int, input().split()) A = list(map(int, input().split())) print(main(N, K, A))