結果

問題 No.1765 While Shining
ユーザー tonyu0
提出日時 2022-04-19 10:57:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 16 ms / 2,000 ms
コード長 1,306 bytes
コンパイル時間 758 ms
コンパイル使用メモリ 101,256 KB
最終ジャッジ日時 2025-01-28 19:17:38
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

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