一道树状数组的题。
话说题目直接告诉做法是什么鬼?
首先这个题直接暴力是(O(n^2))的,不能通过(评论里说可以?可能数据太水了,建议加强)
考虑优化,首先对于答案里的(max),可以直接通过排序优化掉,即把数据从小到大排序,每次更新答案的时候就直接使用当前的(V)就可以了。
之后看这个式子:(|X_i - X j|)。单独看没有什么特点,但是如果我们对于(i),讨论一下(j),就可以发现规律。显然对于(j),可以分成两类,即({j_1|X_{j1}<X_i })和({ j_2|X_{j2}>X_i })。设当前(i)两边分别有(x_1,x_2)个(j),那么答案就是((x_1 imes X_i- Sigma X_{j1})+(Sigma X_{j2}-x_2 imes X_i))。
显然对于(Sigma X_{j1}),可以用前缀和直接求出。对于(Sigma X_{j2}),只需要用总的减去(Sigma X_{j1})就可以求出。
(x_1,x_2)同理也可以求出
这里选择树状数组。
最后别忘了开longlong,否则100pts->45pts。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 20005;
typedef long long ll;
int n;
ll iAns, iSum[N], iCnt[N];
struct sCow
{
int V, X;
}cow[N];
inline bool cmp(sCow a, sCow b){return a.V < b.V;}
inline int lowbit(int x){return x & -x;}
inline void Add(int x, int k){for(; x <= N; x += lowbit(x)) iSum[x] += k, iCnt[x] += 1;}
inline void Query(int x, ll &ans, int &cnt){ans = 0, cnt = 0;for(; x; x -= lowbit(x)) ans += iSum[x], cnt += iCnt[x];}
int main()
{
scanf("%d", &n);
for(int i = 1; i <= n; i++) scanf("%d%d", &cow[i].V, &cow[i].X);
sort(cow + 1, cow + 1 + n, cmp);
ll S = 0, C = 0;
for(int i = 1; i <= n; i++)
{
ll ans = 0;
int cnt = 0;
Query(cow[i].X, ans, cnt);
iAns += cow[i].V * (cnt * cow[i].X - ans);
iAns += cow[i].V * ((S - ans) - (C - cnt) * cow[i].X);
Add(cow[i].X, cow[i].X);
S += cow[i].X, C +=1;
}
cout << iAns;
}