結果
問題 | No.1007 コイン集め |
ユーザー |
![]() |
提出日時 | 2020-03-06 22:26:51 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 85 ms / 1,500 ms |
コード長 | 446 bytes |
コンパイル時間 | 108 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 21,668 KB |
最終ジャッジ日時 | 2024-10-14 08:04:42 |
合計ジャッジ時間 | 1,974 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
import sys input = sys.stdin.readline N, K = map(int, input().split()) a = list(map(int, input().split())) left, right = 0, 0 for i in range(K, N): if a[i] < 2: right += a[i] == 1 break right += a[i] for i in range(K-2, -1, -1): if a[i] < 2: left += a[i] == 1 break left += a[i] if a[K-1] == 0: print(0) elif a[K-1] == 1: print(max(left, right)+1) else: print(left + right + a[K-1])