結果

問題 No.1007 コイン集め
ユーザー KnotsKnots
提出日時 2020-04-16 22:41:23
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,042 bytes
コンパイル時間 1,542 ms
コンパイル使用メモリ 163,600 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-04-10 11:57:21
合計ジャッジ時間 2,590 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 2 ms
6,940 KB
testcase_02 WA -
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 1 ms
6,944 KB
testcase_08 WA -
testcase_09 AC 1 ms
6,940 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 AC 40 ms
6,940 KB
testcase_13 AC 40 ms
6,944 KB
testcase_14 AC 38 ms
6,944 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 42 ms
6,944 KB
testcase_19 AC 44 ms
6,940 KB
testcase_20 AC 45 ms
6,944 KB
testcase_21 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;

const int INF = 1e9;
const int MOD = 1e9+7;
const ll LINF = 1e18;

#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n) for(int i=0;i<(n);++i)
#define REPR(i,n) for(int i=n;i>=0;i--)
#define ALL(v) (v.begin(),v.end())
#define COUT(x) cout<<(x)<<'\n'

template<class T> inline bool chmin(T& a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T& a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}
int main(){
    int n,k;
    cin >> n >> k;
    int a[n+1];
    ll ans = 0;
    REP(i,n){
        cin >>a[i+1];
    }
    ll num_r = 0,num_l=0;
    for(int i=k+1;i<n+1;i++){
        if(a[i]>1)num_r += a[i];
        else break;
    }
    for(int i=k-1;i>0;i--){
        if(a[i]>1)num_l += a[i];
        else break;
    }
    if(a[k]==0)ans = 0;
    else if(a[k]==1)ans = max(num_r,num_l)+1;
    else ans = num_r+num_l+a[k];
    
    COUT(ans);
    return 0;
}
0