結果

問題 No.1007 コイン集め
ユーザー wonda_t_coffeewonda_t_coffee
提出日時 2020-03-06 22:58:09
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 15 ms / 1,500 ms
コード長 1,391 bytes
コンパイル時間 677 ms
コンパイル使用メモリ 81,976 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-14 09:25:52
合計ジャッジ時間 1,711 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <ctime>
#include <cstring>
#include <functional>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

#define rep(i,n)   for (int i = 0; i < (n); ++i)
#define sort(a)    sort((a).begin(), (a).end());
#define uniq(a)    SORT(a);(a).erase(unique((a).begin(), (a).end()), (a).end());
#define find(a,x)  find((a).begin(), (a).end(), (x)) != (a).end()
#define reverse(a) reverse((a).begin(), (a).end());
#define out(d) cout << (d);
#define outl(d) std::cout<<(d)<<"\n";
#define Yes() printf("Yes\n");
#define No() printf("No\n");
#define YES() printf("YES\n");
#define NO() printf("NO\n");
#define ceil(x, y) ((x + y - 1) / (y))

using namespace std;
using ll = long long;
using P = pair<ll,ll>;

const ll MOD = 1000000007; // 10^9 + 7

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  srand((unsigned)time(NULL));

  ll n, k; cin >> n >> k;
  vector<ll> a(n); rep(i, n) cin >> a[i];

  if (a[k - 1] == 0) {
    outl(0)
    return 0;
  }

  ll lsum = 0, rsum = 0;
  for (int i = k - 2; i >= 0; i--) {
    lsum += a[i];
    if (a[i] <= 1) break;
  }
  for (int i = k; i < n; i++) {
    rsum += a[i];
    if (a[i] <= 1) break;
  }

  if (a[k - 1] == 1) {
    outl(max(lsum, rsum) + 1)
  } else {
    outl(lsum + rsum + a[k - 1])
  }
}
0