結果
問題 |
No.1007 コイン集め
|
ユーザー |
|
提出日時 | 2025-09-07 22:23:44 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 580 bytes |
コンパイル時間 | 388 ms |
コンパイル使用メモリ | 82,384 KB |
実行使用メモリ | 88,532 KB |
最終ジャッジ日時 | 2025-09-07 22:23:47 |
合計ジャッジ時間 | 2,526 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 WA * 1 |
ソースコード
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,r0): if A[i]==1: r1 = i break right_sum = sum(A[K+1:r1]) if r1<N+1: 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)