結果

問題 No.1007 コイン集め
ユーザー Mishan5055Mishan5055
提出日時 2020-03-09 02:17:26
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,322 bytes
コンパイル時間 1,537 ms
コンパイル使用メモリ 168,988 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-07 20:26:59
合計ジャッジ時間 2,782 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

diff #

#include <bits/stdc++.h>
#include <iostream>
using namespace std;
int main(void){
    int N,K;
    cin >> N >> K;
    vector<long> A(N);
    vector<long> ASUM(N);
    
    int startidx=0;
    int endidx=N-1;
    
    for(int i=0;i<N;i++){
        if(i!=K-1){
            long s;
            cin >> s;
            A.at(i)=s;
            if(i==0){
                ASUM.at(i)=s;
            }else{
                ASUM.at(i)=ASUM.at(i-1)+s;
            }
            if(s<=1&&i<K-1){
                startidx=i;
            }else if(s<=1&&i>K-1){
                endidx=i;
            }
        }else{
            cin >> A.at(i);
            if(i>0){
                ASUM.at(i)=ASUM.at(i-1)+A.at(i);
            }else{
                ASUM.at(i)=A.at(i);
            }
        }
    }

    if(A.at(K-1)==0){
        cout << 0 << endl;
    }else if(A.at(K-1)==1){
        long ans1=ASUM.at(K-1)-ASUM.at(startidx);
        if(startidx==0){
            ans1+=ASUM.at(0);
        }
        long ans2=ASUM.at(endidx)-ASUM.at(K-1);
        if(ans1<ans2){
            cout << ans2 << endl;
        }else{
            cout << ans1 << endl;
        }
    }else{
        if(startidx>0){
            cout << ASUM.at(endidx)-ASUM.at(startidx-1) << endl;
        }else{
            cout << ASUM.at(endidx) << endl;
        }
    }
    
}
0