結果

問題 No.1007 コイン集め
ユーザー flippergo
提出日時 2025-09-07 22:31:32
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 81 ms / 1,500 ms
コード長 579 bytes
コンパイル時間 458 ms
コンパイル使用メモリ 82,632 KB
実行使用メモリ 87,296 KB
最終ジャッジ日時 2025-09-07 22:31:36
合計ジャッジ時間 3,025 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

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 = l0
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 = r0
for i in range(K+1,r0):
    if A[i]==1:
        r1 = i
        break
right_sum = sum(A[K+1:r1])
if r1<r0:
    right_sum += 1
ans = A[K]
if A[K]==1:
    ans += max(left_sum,right_sum)
elif A[K]>1:
    ans += left_sum+right_sum
print(ans)
0