chenfy27的刷题记录

导航

abc351F Double Sum

给定数组A[N],对所有1<=i<j<=N,计算max(A[j]-A[i],0)之和。
2<=N<=4E5; 0<=A[i]<=1E8

分析:从左到后依次处理,用平衡树维护左侧A[i],对于A[j],只需要统计值小于A[j]的那些A[i]即可,可以合并求和过程转化为前缀和。

#include <bits/stdc++.h>
using i64 = long long;

// SumTreap模板...

void solve() {
    int N;
    std::cin >> N;
    SumTreap<i64> tr;
    std::vector<int> A(N);
    for (int i = 0; i < N; i++) {
        std::cin >> A[i];
    }

    i64 ans = 0;
    for (int i = 0; i < N; i++) {
        i64 cnt1 = tr.ltcnt(A[i]);
        ans += cnt1 * A[i] - tr.ksum(cnt1);
        tr.insert(A[i]);
    }
    std::cout << ans << "\n";
}

int main() {
    std::cin.tie(0)->sync_with_stdio(0);
    int t = 1;
    while (t--) solve();
    return 0;
}

posted on 2024-10-08 20:13  chenfy27  阅读(13)  评论(0)    收藏  举报