結果

問題 No.1765 While Shining
ユーザー tonyu0tonyu0
提出日時 2022-04-19 10:57:24
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 1,306 bytes
コンパイル時間 875 ms
コンパイル使用メモリ 105,236 KB
実行使用メモリ 4,384 KB
最終ジャッジ日時 2023-08-30 06:21:50
合計ジャッジ時間 2,274 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,384 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 1 ms
4,376 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 2 ms
4,380 KB
testcase_09 AC 10 ms
4,384 KB
testcase_10 AC 14 ms
4,376 KB
testcase_11 AC 13 ms
4,380 KB
testcase_12 AC 14 ms
4,376 KB
testcase_13 AC 12 ms
4,380 KB
testcase_14 AC 15 ms
4,376 KB
testcase_15 AC 15 ms
4,380 KB
testcase_16 AC 15 ms
4,380 KB
testcase_17 AC 15 ms
4,376 KB
testcase_18 AC 15 ms
4,380 KB
testcase_19 AC 15 ms
4,376 KB
testcase_20 AC 15 ms
4,376 KB
testcase_21 AC 15 ms
4,384 KB
testcase_22 AC 15 ms
4,380 KB
testcase_23 AC 15 ms
4,376 KB
testcase_24 AC 13 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cmath>
#include <iostream>
#include <numeric>
#include <queue>
#include <unordered_map>
#include <vector>
using namespace std;
using ll = long long;
#define rep(i, j, n) for (int i = j; i < (n); ++i)
#define rrep(i, j, n) for (int i = (n)-1; j <= i; --i)
#define all(a) a.begin(), a.end()
template <typename T>
std::ostream &operator<<(std::ostream &os, std::vector<T> &a) {
  for (size_t i = 0; i < a.size(); ++i) os << (i > 0 ? " " : "") << a[i];
  return os << '\n';
}
template <typename T>
std::istream &operator>>(std::istream &is, std::vector<T> &a) {
  for (T &x : a) { is >> x; }
  return is;
}
[[maybe_unused]] constexpr long long MOD = 998244353;
// constexpr int MOD = 1000000007;
[[maybe_unused]] constexpr int INF = 0x3f3f3f3f;
[[maybe_unused]] constexpr long long INFL = 0x3f3f3f3f3f3f3f3fLL;

int main() {
  cin.tie(0)->sync_with_stdio(0);
  int n;
  cin >> n;
  vector<int> a(n);
  cin >> a;
  ll ans = 0;
  // 01010101となる範囲の1の数、区間の長さから足す値が決まる
  rep(i, 0, n - 1) {
    if (i < n - 1 && a[i] == 1) {
      int j = i;
      while (j < n - 2 && a[j] != a[j + 1]) ++j;
      ll len = j - i + 1;
      ll m = (len + 2) / 2;
      ans += (len + 2) * m - m * (m + 1);
      i = j;
    }
  }
  cout << ans;
  return 0;
}
0