結果

問題 No.1007 コイン集め
ユーザー convexineq
提出日時 2020-03-06 22:02:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 63 ms / 1,500 ms
コード長 528 bytes
コンパイル時間 166 ms
コンパイル使用メモリ 82,528 KB
実行使用メモリ 80,812 KB
最終ジャッジ日時 2024-10-14 06:58:01
合計ジャッジ時間 2,249 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

# coding: utf-8
# Your code here!

import sys
readline = sys.stdin.readline
read = sys.stdin.read

n,k,*a = [int(i) for i in read().split()]
k -= 1

v = 0
rb = 0
for i in range(k+1,n):
    v += a[i]
    if a[i] <= 1:
        rb = v
        break
else:
    rb = v
        
v = 0
lb = 0
for i in range(k-1,-1,-1):
    v += a[i]
    if a[i] <= 1:
        lb = v
        break
else:
    lb = v

#print(r,rb)
#print(l,lb)

if a[k] == 0:
    print(0)
elif a[k] == 1:
    print(a[k]+max(lb,rb))
else:
    print(a[k]+lb+rb)
0