結果

問題 No.1505 Zero-Product Ranges
ユーザー ooooobaoooooba
提出日時 2021-07-31 10:40:00
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 39 ms / 2,000 ms
コード長 860 bytes
コンパイル時間 1,303 ms
コンパイル使用メモリ 116,928 KB
最終ジャッジ日時 2025-01-23 12:59:09
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <array>
#include <cmath>
#include <cstdio>
#include <deque>
#include <iomanip>
#include <iostream>
#include <numeric>
#include <optional>
#include <queue>
#include <set>
#include <stack>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std;

int main() {
    int32_t n;
    cin >> n;
    vector<bool> as(n);
    for (auto &&a : as) {
        int32_t x;
        cin >> x;
        a = x == 1;
    }
    vector<int32_t> rl;
    int32_t cnt = 0;
    for (auto a : as) {
        if (a)
            ++cnt;
        else {
            rl.push_back(cnt);
            cnt = 0;
        }
    }
    rl.push_back(cnt);
    for (auto &&x : rl) {
        ++x;
    }
    int64_t ans = 0, sum = 0;
    for (auto x : rl) {
        ans += x * sum;
        sum += x;
    }
    cout << ans << endl;
    return 0;
}
0