#439851 (C++11) No.1007 コイン集め

• 問題: No.1007 コイン集め
• ユーザー: Y17
• 提出日時: 2020-03-06 22:37:35
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 47 ms / 1,500 ms
• コード長: 778 bytes
• コンパイル時間: 1,991 ms
• コンパイル使用メモリ: 163,584 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-10-14 08:58:37
• 合計ジャッジ時間: 3,063 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,816 KB
testcase_01	AC	1 ms; 6,816 KB
testcase_02	AC	1 ms; 6,816 KB
testcase_03	AC	2 ms; 6,816 KB
testcase_04	AC	2 ms; 6,820 KB
testcase_05	AC	2 ms; 6,820 KB
testcase_06	AC	1 ms; 6,816 KB
testcase_07	AC	2 ms; 6,816 KB
testcase_08	AC	2 ms; 6,816 KB
testcase_09	AC	2 ms; 6,816 KB
testcase_10	AC	2 ms; 6,820 KB
testcase_11	AC	1 ms; 6,816 KB
testcase_12	AC	44 ms; 6,816 KB
testcase_13	AC	44 ms; 6,824 KB
testcase_14	AC	43 ms; 6,816 KB
testcase_15	AC	18 ms; 6,820 KB
testcase_16	AC	17 ms; 6,816 KB
testcase_17	AC	18 ms; 6,816 KB
testcase_18	AC	47 ms; 6,816 KB
testcase_19	AC	47 ms; 6,820 KB
testcase_20	AC	47 ms; 6,820 KB
testcase_21	AC	46 ms; 6,816 KB

ソースコード

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

#define int long long

int n, k;
int a[100010];

signed main(){
	cin >> n >> k;

	for(int i = 0;i < n;i++){
		cin >> a[i];
	}
	k--;

	vector<int> lv, rv;
	int ans = 0;

	for(int i = 0;i < k;i++){
		if(a[i] <= 1) lv.clear();
		if(a[i] > 0) lv.push_back(a[i]);
	}

	int left = 0;
	int right = 0;
	int noko = 0;
	if(lv.size() > 0){
		for(int i = 0;i < lv.size();i++){
			left += lv[i];
		}
	}

	for(int i = n-1;i > k;i--){
		if(a[i] <= 1) rv.clear();
		if(a[i] > 0) rv.push_back(a[i]);
	}

	if(rv.size() > 0){
		for(int i = 0;i < rv.size();i++){
			right += rv[i];
		}
	}

	if(a[k] == 0){
		cout << 0 << endl;
	}else if(a[k] == 1){
		cout << max(left, right)+1 << endl;
	}else{
		cout << left + right + a[k] << endl;
	}

	return 0;
}