結果
問題 | No.1007 コイン集め |
ユーザー |
![]() |
提出日時 | 2020-03-06 22:38:54 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 76 ms / 1,500 ms |
コード長 | 651 bytes |
コンパイル時間 | 95 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 21,564 KB |
最終ジャッジ日時 | 2024-10-14 08:59:43 |
合計ジャッジ時間 | 2,010 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
import sys sys.setrecursionlimit(10 ** 8) input = sys.stdin.readline def main(): N, K = [int(x) for x in input().split()] A = [int(x) for x in input().split()] if A[K - 1] == 0: print(0) return ans1 = 0 for a in A[K:]: if a < 2: ans1 += a break ans1 += a ans2 = 0 for i in range(K - 2, -1, -1): a = A[i] if a < 2: ans2 += a break ans2 += a if A[K - 1] == 1: print(max(ans1, ans2) + A[K - 1]) else: print(ans1 + ans2 + A[K - 1]) if __name__ == '__main__': main()