結果
問題 | No.1007 コイン集め |
ユーザー | batsumaru |
提出日時 | 2020-03-06 22:09:27 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 43 ms / 1,500 ms |
コード長 | 928 bytes |
コンパイル時間 | 1,494 ms |
コンパイル使用メモリ | 168,668 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-14 07:13:27 |
合計ジャッジ時間 | 2,877 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<ll, ll> P; #define DUMP(x) cout << #x << " = " << (x) << endl; #define FOR(i, m, n) for (ll i = m; i < n; i++) #define IFOR(i, m, n) for (ll i = n - 1; i >= m; i--) #define REP(i, n) FOR(i, 0, n) #define IREP(i, n) IFOR(i, 0, n) #define FOREACH(x, a) for (auto&(x) : (a)) #define ALL(v) (v).begin(), (v).end() #define SZ(x) ll(x.size()) int main() { ll n, k; cin >> n >> k; k--; vector<ll> a(n), s(n+1,0); REP(i,n){ cin >> a[i]; s[i+1] = s[i] + a[i]; } if(a[k] == 0){ cout << 0 << endl; return 0; } ll l = 0, r = n-1; IREP(i,k){ if(a[i] <= 1){ l = i; break; } if(i==0){ l = 0; } } FOR(i,k+1,n){ if(a[i] <= 1){ r = i; break; } } if(a[k]>=2){ cout << s[r+1] - s[l] << endl; return 0; } cout << max(s[k+1]-s[l], s[r+1]-s[k]) << endl; }