結果

問題 No.1505 Zero-Product Ranges
ユーザー yansi819
提出日時 2024-03-22 10:09:59
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 38 ms / 2,000 ms
コード長 531 bytes
コンパイル時間 4,146 ms
コンパイル使用メモリ 250,664 KB
最終ジャッジ日時 2025-02-20 10:53:16
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long;
using ld = long double;

int main() {
  int N; cin >> N;
  vector<int> A(N + 1);
  for (int i = 1; i <= N; i++) cin >> A[i];
  ll ans = 0;
  for (int i = 1; i <= N; i++) ans += i;
  ll cnt = 0;
  for (int i = 1; i <= N; i++) {
    if (A[i] == 1) cnt++;
    else {
      for (int j = 1; j <= cnt; j++) ans -= j;
      cnt = 0;
    }
  }
  for (int j = 1; j <= cnt; j++) ans -= j;
  cout << ans << endl;
  return 0;
}
0