結果

問題 No.2956 Substitute with Average
ユーザー Xinyuan Huang
提出日時 2025-02-25 22:48:58
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 570 ms / 3,000 ms
コード長 878 bytes
コンパイル時間 3,637 ms
コンパイル使用メモリ 281,932 KB
実行使用メモリ 12,352 KB
最終ジャッジ日時 2025-02-25 22:49:14
合計ジャッジ時間 14,358 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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);
        nzero.reserve(n);
        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