結果
問題 | No.1007 コイン集め |
ユーザー |
![]() |
提出日時 | 2020-03-06 22:48:47 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 46 ms / 1,500 ms |
コード長 | 1,057 bytes |
コンパイル時間 | 1,423 ms |
コンパイル使用メモリ | 168,540 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-10-14 09:15:28 |
合計ジャッジ時間 | 2,577 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define REP(i,a,n) for(int i=(a); i<(int)(n); i++) #define rep(i,n) REP(i,0,n) #define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it) #define ALLOF(c) (c).begin(), (c).end() typedef long long ll; typedef unsigned long long ull; int main(){ int N, K; cin >> N >> K; vector<int> v; rep(i,N){ int a; cin >> a; v.push_back(a); } K--; ll lhs = 0, rhs = 0; { int i = K-1; while(i>=0){ if(v[i] >= 2){ lhs += v[i]; } else if(v[i] == 1){ lhs += 1; break; } else { break; } i--; } } { int i = K+1; while(i<N){ if(v[i] >= 2){ rhs += v[i]; } else if(v[i] == 1){ rhs += 1; break; } else { break; } i++; } } if(v[K] == 0){ cout << 0 << endl; } else if(v[K] == 1){ cout << max(lhs, rhs) + 1 << endl; } else { cout << lhs + rhs + v[K] << endl; } return 0; }