結果
問題 |
No.1007 コイン集め
|
ユーザー |
|
提出日時 | 2025-09-07 22:17:30 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 585 bytes |
コンパイル時間 | 450 ms |
コンパイル使用メモリ | 82,308 KB |
実行使用メモリ | 94,992 KB |
最終ジャッジ日時 | 2025-09-07 22:17:34 |
合計ジャッジ時間 | 3,300 ms |
ジャッジサーバーID (参考情報) |
judge / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 13 WA * 1 RE * 5 |
ソースコード
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 = 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)