結果
問題 | No.1007 コイン集め |
ユーザー |
![]() |
提出日時 | 2021-01-14 01:02:51 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 14 ms / 1,500 ms |
コード長 | 2,998 bytes |
コンパイル時間 | 1,776 ms |
コンパイル使用メモリ | 168,480 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-22 21:29:38 |
合計ジャッジ時間 | 2,968 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#include <bits/stdc++.h>#define rep(i, n) for(int(i) = 0; (i) < (n); (i)++)#define FOR(i, m, n) for(int(i) = (m); (i) < (n); (i)++)#define ALL(v) (v).begin(), (v).end()#define LLA(v) (v).rbegin(), (v).rend()#define SZ(v) (v).size()#define INT(...) \int __VA_ARGS__; \read(__VA_ARGS__)#define LL(...) \ll __VA_ARGS__; \read(__VA_ARGS__)#define DOUBLE(...) \double __VA_ARGS__; \read(__VA_ARGS__)#define CHAR(...) \char __VA_ARGS__; \read(__VA_ARGS__)#define STRING(...) \string __VA_ARGS__; \read(__VA_ARGS__)#define VEC(type, name, size) \vector<type> name(size); \read(name)using namespace std;using ll = long long;using pii = pair<int, int>;using pll = pair<ll, ll>;using Graph = vector<vector<int>>;template <typename T> struct edge {int to;T cost;edge(int t, T c) : to(t), cost(c) {}};template <typename T> using WGraph = vector<vector<edge<T>>>;const int INF = 1 << 30;const ll LINF = 1LL << 60;const int MOD = 1e9 + 7;template <class T> inline vector<T> make_vec(size_t a, T val) {return vector<T>(a, val);}template <class... Ts> inline auto make_vec(size_t a, Ts... ts) {return vector<decltype(make_vec(ts...))>(a, make_vec(ts...));}void read() {}template <class T> inline void read(T &a) { cin >> a; }template <class T, class S> inline void read(pair<T, S> &p) {read(p.first), read(p.second);}template <class T> inline void read(vector<T> &v) {for(auto &a : v)read(a);}template <class Head, class... Tail>inline void read(Head &head, Tail &... tail) {read(head), read(tail...);}template <class T> inline void chmax(T &a, T &b) { (a < b ? a = b : a); }template <class T> inline void chmin(T &a, T &b) { (a > b ? a = b : a); }int main() {ios::sync_with_stdio(false);cin.tie(nullptr);INT(n, k);k--;VEC(ll, a, n);ll left = 0, right = 0;for(int i = k - 1; i >= 0; i--) {left += a[i];if(a[i] <= 1)break;}for(int i = k + 1; i < n; i++) {right += a[i];if(a[i] <= 1)break;}if(a[k] == 0) {cout << 0 << endl;} else if(a[k] == 1) {cout << max(left, right) + 1 << endl;} elsecout << left + right + a[k] << endl;}