結果

問題 No.2956 Substitute with Average
ユーザー Xinyuan Huang
提出日時 2025-02-25 22:46:55
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,918 ms / 3,000 ms
コード長 901 bytes
コンパイル時間 3,403 ms
コンパイル使用メモリ 281,340 KB
実行使用メモリ 111,576 KB
最終ジャッジ日時 2025-02-25 22:47:57
合計ジャッジ時間 35,373 ms
ジャッジサーバーID
（参考情報） 		judge7 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using i64 = long long;

constexpr int M = 30;

int main() {
    int n; std::cin >> n;
    std::vector<int> a(n + 1);
    for (int i = 1; i <= n; ++i)
        std::cin >> a[i];
    i64 res = n * (n + 1LL) / 2;
    for (int x = 1; x <= M; ++x) {
        std::vector<int> b(n + 1), s(n + 1);
        for (int i = 1; i <= n; ++i)
            b[i] = a[i] - x;
        std::partial_sum(b.begin(), b.end(), s.begin());
        std::unordered_map<int, int> zero, nzero;
        zero.reserve(n * M * 2 + 1);
        nzero.rehash(n * M * 2 + 1);
        for (int i = 1; i <= n; ++i) {
            if (b[i] == 0) {
                zero[s[i - 1]] += 1;
                res -= zero[s[i]] + nzero[s[i]];
            }
            else {
                nzero[s[i - 1]] += 1;
                res -= zero[s[i]];
            }
        }
    }
    std::cout << res + 1 << "\n";
    return 0;
}
0