結果

問題 No.1007 コイン集め
ユーザー convexineqconvexineq
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,268 KB
testcase_01 AC 37 ms
53,032 KB
testcase_02 AC 37 ms
53,360 KB
testcase_03 AC 38 ms
52,584 KB
testcase_04 AC 38 ms
52,476 KB
testcase_05 AC 38 ms
52,644 KB
testcase_06 AC 37 ms
52,364 KB
testcase_07 AC 37 ms
52,676 KB
testcase_08 AC 38 ms
53,280 KB
testcase_09 AC 37 ms
52,756 KB
testcase_10 AC 37 ms
53,252 KB
testcase_11 AC 37 ms
52,356 KB
testcase_12 AC 63 ms
80,812 KB
testcase_13 AC 62 ms
79,152 KB
testcase_14 AC 63 ms
79,728 KB
testcase_15 AC 55 ms
74,000 KB
testcase_16 AC 56 ms
74,676 KB
testcase_17 AC 55 ms
74,480 KB
testcase_18 AC 61 ms
78,240 KB
testcase_19 AC 62 ms
77,360 KB
testcase_20 AC 62 ms
78,524 KB
testcase_21 AC 61 ms
77,776 KB
権限があれば一括ダウンロードができます

ソースコード

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