結果
問題 | No.1007 コイン集め |
ユーザー |
![]() |
提出日時 | 2020-03-06 21:49:39 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 15 ms / 1,500 ms |
コード長 | 840 bytes |
コンパイル時間 | 1,680 ms |
コンパイル使用メモリ | 168,144 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-14 06:13:42 |
合計ジャッジ時間 | 2,391 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#include <bits/stdc++.h> using namespace std; using lint = long long; const lint inf = 1LL << 60; const lint mod = 1000000007; int main() { cin.tie(nullptr); ios::sync_with_stdio(false); lint n, k; cin >> n >> k; k--; vector<int> a(n); for (int i = 0; i < n; ++i) { cin >> a[i]; } lint ret = a[k]; if (a[k] == 0) { cout << 0 << "\n"; return 0; } lint retr = 0; for (int i = k + 1; i < n; ++i) { retr += a[i]; if (a[i] == 1 || a[i] == 0) break; } lint retl = 0; for (int i = k - 1; i >= 0; --i) { retl += a[i]; if (a[i] == 1 || a[i] == 0) break; } if (a[k] == 1) { ret += max(retl, retr); } else { ret += retl + retr; } cout << ret << "\n"; return 0; }