Submission #69320905

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
int main(void){
	int N,R;
	cin >> N >> R;
	vector<int> L(N);
	for(int i=0;i<N;i++){
		cin >> L[i];
	}
	int start = 0;
	for(int i=0;i<N;i++){
		if ( L[i] == 0 ) {
			start = i;
			break;
		}
	}
	start = std::min(start,R);
	int end = N-1;
	for(int i=N-1;i>=0;i--){
		if ( L[i] == 0 ) {
			end = i;
			break;
		}
	}
	end = std::max(end,R-1);
	long ans = 0;
	for(int i=start;i<=end;i++){
		if ( L[i] == 0 ) {
			ans += 1;
		}
		else {
			ans += 2;
		}
	}
	cout << ans << endl;
	return 0;
}

Submission Info

Judge Result