結果

問題 No.1007 コイン集め
コンテスト
ユーザー 学ぶマン
提出日時 2025-12-03 17:57:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 63 ms / 1,500 ms
コード長 568 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 378 ms
コンパイル使用メモリ 82,280 KB
実行使用メモリ 84,816 KB
最終ジャッジ日時 2025-12-03 17:57:30
合計ジャッジ時間 2,642 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N, K = map(int, input().split())
A = list(map(int, input().split()))
K -= 1

if A[K] == 0:
    exit(print(0))

# 0 or 1 にたどりつくまで左右に探索
left = 0
for i in range(K - 1, -1, -1):
    if A[i] == 0:
        break
    if A[i] == 1:
        left += 1
        break
    left += A[i]

right = 0
for i in range(K + 1, N):
    if A[i] == 0:
        break
    if A[i] == 1:
        right += 1
        break
    right += A[i]

if A[K] == 1: # 左右どちらかだけいける
    ans = max(left, right) + 1
else:
    ans = left + right + A[K]

print(ans)
0