結果

問題 No.2956 Substitute with Average
コンテスト
ユーザー nono00
提出日時 2024-11-09 00:02:03
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,188 bytes
コンパイル時間 3,024 ms
コンパイル使用メモリ 248,264 KB
実行使用メモリ 100,924 KB
最終ジャッジ日時 2024-11-09 00:02:11
合計ジャッジ時間 7,873 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 4 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
using ull = unsigned long long;
template <class T>
using max_heap = priority_queue<T>;
template <class T>
using min_heap = priority_queue<T, vector<T>, greater<T>>;
#define rep(i, l, r) for (ll i = (l); i < (r); i++)
#define rrep(i, r, l) for (ll i = (r); i-- > (l);)
#define all(x) begin(x), end(x)
template <class T>
bool chmin(T& lhs, T rhs) {
    return lhs > rhs ? (lhs = rhs, true) : false;
}
template <class T>
bool chmax(T& lhs, T rhs) {
    return lhs < rhs ? (lhs = rhs, true) : false;
}
struct IOIO {
    IOIO() {
        cin.tie(0)->sync_with_stdio(0);
    }
} ioio;
template <class S, class T>
ostream& operator<<(ostream& os, const pair<S, T>& p) {
    os << '(' << p.first << ", " << p.second << ')';
    return os;
}
template <class T>
ostream& operator<<(ostream& os, const vector<T>& vs) {
    os << '{';
    rep(i, 0, (int)vs.size()) os << vs[i] << (i + 1 == (int)vs.size() ? "" : ", ");
    os << '}';
    return os;
}
template <class T>
ostream& operator<<(ostream& os, const set<T>& vs) {
    os << '{';
    for (auto it = vs.begin(); it != vs.end(); it++) {
        if (it != vs.begin()) {
            os << ", ";
        }
        os << *it;
    }
    os << '}';
    return os;
}
template <class T, class U>
ostream& operator<<(ostream& os, const map<T, U>& vs) {
    os << '{';
    for (auto it = vs.begin(); it != vs.end(); it++) {
        if (it != vs.begin()) {
            os << ", ";
        }
        os << *it;
    }
    os << '}';
    return os;
}
#ifdef DEBUG
void dump_func() {
    cerr << endl;
}
template <class Head, class... Tail>
void dump_func(Head&& head, Tail&&... tail) {
    cerr << head;
    if (sizeof...(Tail) > 0) {
        cerr << ", ";
    }
    dump_func(std::move(tail)...);
}
#define dump(...) cerr << "[" + string(#__VA_ARGS__) + "] ", dump_func(__VA_ARGS__)
#else
#define dump(...) static_cast<int>(0)
#endif

constexpr int N = 2 * 100000 * 32;
constexpr int OFFSET = 100000 * 32;
int MP[31][N];
int SUMS[31];

void solve() {
    int n;
    cin >> n;
    vector<int> a(n);
    rep(i, 0, n) cin >> a[i];
    ll ans = 0;
    auto DnC = [&](auto&& self, int l, int r) -> void {
        if (l + 1 == r) return;
        int m = (l + r) / 2;
        self(self, l, m);
        self(self, m, r);
        rep(x, 1, 31) {
            int sum = 0;
            rep(i, m, r) {
                sum += a[i];
                if (x != a[i]) {
                    MP[x][(i - m + 1) * x - sum + OFFSET]++;
                    SUMS[x]++;
                }
            }
        }
        {
            int sum = 0;
            rrep(i, m, l) {
                sum += a[i];
                ans += SUMS[a[i]] - MP[a[i]][sum - a[i] * (m - i) + OFFSET];
            }
        }
        rep(x, 1, 31) {
            int sum = 0;
            rep(i, m, r) {
                sum += a[i];
                if (x != a[i]) {
                    MP[x][(i - m + 1) * x - sum + OFFSET]--;
                    SUMS[x]--;
                }
            }
        }
    };
    DnC(DnC, 0, n);
    cout << ans + 1 << '\n';
}

int main() {
    int t = 1;

    while (t--) solve();
}
0