結果

問題 No.1007 コイン集め
ユーザー flippergo
提出日時 2025-09-07 22:13:36
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 585 bytes
コンパイル時間 366 ms
コンパイル使用メモリ 82,572 KB
実行使用メモリ 94,668 KB
最終ジャッジ日時 2025-09-07 22:13:40
合計ジャッジ時間 3,565 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 1 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff #

N,K = map(int,input().split())
A = [0]+list(map(int,input().split()))+[0]
l0 = 0
for i in range(K-1,0,-1):
    if A[i]==0:
        l0 = i
        break
l1 = 0
for i in range(K-1,l0,-1):
    if A[i]==1:
        l1 = i
        break
left_sum = sum(A[l1:K])
r0 = N+1
for i in range(K+1,N+1):
    if A[i]==0:
        r0 = i
        break
r1 = N+1
for i in range(K+1,r1):
    if A[i]==1:
        r1 = i
        break
right_sum = sum(A[K+1:r1])
if r1<N+1:
    right_sum += 1
ans = 0
if A[K]==1:
    ans = A[k]+max(left_sum,right_sum)
elif A[K]>1:
    ans = A[K]+left_sum+right_sum
print(ans)
0