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;
}
浙公网安备 33010602011771号