結果
問題 | No.1007 コイン集め |
ユーザー | uni_python |
提出日時 | 2020-08-10 23:25:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 63 ms / 1,500 ms |
コード長 | 888 bytes |
コンパイル時間 | 387 ms |
コンパイル使用メモリ | 82,152 KB |
実行使用メモリ | 82,304 KB |
最終ジャッジ日時 | 2024-10-08 20:25:52 |
合計ジャッジ時間 | 2,662 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
import sys input=sys.stdin.readline def I(): return int(input()) def MI(): return map(int, input().split()) def LI(): return list(map(int, input().split())) def main(): mod=10**9+7 N,K=MI() K-=1 A=LI() if A[K]==0: print(0) exit() if min(A)>=2: print(sum(A)) else: ans=A[K] aa=0 bb=0 for i in range(K+1,N): if A[i]==1: aa+=1 break elif A[i]==0: break else: aa+=A[i] for i in range(K-1,-1,-1): if A[i]==1: bb+=1 break elif A[i]==0: break else: bb+=A[i] if A[K]==1: ans+=max(aa,bb) else: ans+=aa+bb print(ans) main()