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