Submission #69370868

---

Source Code Expand

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

/////////////////// メイン ///////////////////

int main () {
  
  //////////////////// 入力 ////////////////////

  int n, r;
  cin >> n >> r;

  vector<int> l(n);
  for (int i=0; i<n; i++) {
    cin >> l.at(i);
  }

  //////////////// 出力変数定義 ////////////////

  int result = 0;

  //////////////////// 処理 ////////////////////

  // ここより左は閉めっぱなしでいい、という境界値のドア番号
  // ただし、高橋君の初期位置より左しか考えない
  int left = r;

  // ここより右は閉めっぱなしでいい、という境界値のドア番号
  // ただし、高橋君の初期位置より右しか調べない
  int right = r-1;
  
  // ドアを1枚ずつ見て、開いていたら情報更新
  for (int i=0; i<n; i++) {
    if (l.at(i)==0) {
      left = min(left,i);
      right = max(right,i);
    }
  }

  // 必要な区間のドアを全部開ける
  for (int i=left; i<=right; i++) {
    result += l.at(i);
  }

  // そして順に全部閉める
  result += right-left+1;

  //////////////////// 出力 ////////////////////

  cout << result << endl;

  //////////////////// 終了 ////////////////////

  return 0;

}

Submission Info

Judge Result