結果
問題 | No.1007 コイン集め |
ユーザー | chielo |
提出日時 | 2020-03-06 22:31:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 93 ms / 1,500 ms |
コード長 | 719 bytes |
コンパイル時間 | 274 ms |
コンパイル使用メモリ | 86,992 KB |
実行使用メモリ | 91,480 KB |
最終ジャッジ日時 | 2023-08-04 11:19:05 |
合計ジャッジ時間 | 3,070 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 64 ms
71,284 KB |
testcase_01 | AC | 65 ms
71,312 KB |
testcase_02 | AC | 64 ms
70,984 KB |
testcase_03 | AC | 66 ms
71,124 KB |
testcase_04 | AC | 69 ms
71,256 KB |
testcase_05 | AC | 69 ms
70,968 KB |
testcase_06 | AC | 67 ms
71,228 KB |
testcase_07 | AC | 67 ms
71,384 KB |
testcase_08 | AC | 67 ms
71,472 KB |
testcase_09 | AC | 65 ms
71,468 KB |
testcase_10 | AC | 68 ms
70,980 KB |
testcase_11 | AC | 66 ms
71,260 KB |
testcase_12 | AC | 91 ms
91,052 KB |
testcase_13 | AC | 89 ms
91,100 KB |
testcase_14 | AC | 88 ms
91,008 KB |
testcase_15 | AC | 85 ms
88,596 KB |
testcase_16 | AC | 84 ms
88,344 KB |
testcase_17 | AC | 86 ms
88,548 KB |
testcase_18 | AC | 93 ms
91,104 KB |
testcase_19 | AC | 88 ms
91,000 KB |
testcase_20 | AC | 90 ms
91,480 KB |
testcase_21 | AC | 89 ms
91,160 KB |
ソースコード
n, k = map(int, input().split()) k -= 1 a = list(map(int, input().split())) if a[k] == 0: print(0) exit() def findlr(k): if k < 0 or k >= n: return 0 if a[k] <= 1: return a[k] l, r = k, k while l - 1 >= 0 and a[l - 1] >= 2: l -= 1 while r + 1 < n and a[r + 1] >= 2: r += 1 if l - 1 >= 0 and a[l - 1] >= 1: l -= 1 if r + 1 < n and a[r + 1] >= 1: r += 1 ans = 0 for i in range(l, r + 1): ans += a[i] return ans if a[k] == 1: if findlr(k - 1) > 1 or findlr(k + 1) > 1: print(max(findlr(k - 1), findlr(k + 1))) else: print(max(1, findlr(k - 1) + 1, findlr(k + 1) + 1)) else: print(findlr(k))