結果

問題 No.1007 コイン集め
ユーザー forest3
提出日時 2020-07-10 11:44:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 563 bytes
コンパイル時間 1,574 ms
コンパイル使用メモリ 168,512 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-10 04:44:52
合計ジャッジ時間 2,725 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
 
int main()
{
	int N, K;
	cin >> N >> K;
	vector<long long> A( N );
	for( int i = 0; i < N; i++ ) {
		cin >> A[i];
	}

	K--;
	long long ans1 = 0;
	for( int i = K + 1; i < N; i++ ) {
		if( A[i] < 1 ) break;
		ans1 += A[i];
		if( A[i] < 2 ) break;
	}
	long long ans2 = 0;
	for( int i = K - 1; i >= 0; i-- ) {
		if( A[i] < 1 ) break;
		ans2 += A[i];
		if( A[i] < 2 ) break;
	}
	long long ans = 0;
	if( A[K] == 1 ) max( ans1 + A[K], ans2 + A[K] );
	else if( A[K] > 1 ) ans = ans1 + ans2 + A[K];

	cout << ans << endl;
}
0