結果

問題 No.1007 コイン集め
ユーザー Knots
提出日時 2020-04-16 22:35:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,035 bytes
コンパイル時間 1,548 ms
コンパイル使用メモリ 165,780 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-02 13:35:59
合計ジャッジ時間 2,770 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 6
権限があれば一括ダウンロードができます

ソースコード

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++){
        num_r += a[i];
        if(a[i]==1)break;
    }
    for(int i=k-1;i>=0;i--){
        num_l += a[i];
        if(a[i]==1)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