题意:横坐标上若干点,坐标为X[i],每个点有个权值val,问你所有两两 m a x ( v a l [ i ] , v a l [ j ] ) ∗ ∣ X [ i ] − X [ j ] ∣ max(val[i],val[j])*|X[i]-X[j]| max(val[i],val[j])∗∣X[i]−X[j]∣的和
思路:
比较裸的cdq分治题目,我们把题意条件分成两部分:
对于当前i要计数的有
1.val[j]比当前val[i]小的(这样val[i]才用得上)。
2.同时要知道他们的坐标来求和。
对于第一点,我们只需要先按照val从小到大排序,这样对于i,满足第一点的已经在前i-1就已经处理过了,现在我们要在这些处理过的地方求
∣
X
[
i
]
−
X
[
j
]
∣
|X[i]-X[j]|
∣X[i]−X[j]∣的和,对于这个,我们可以变一下式,变成
X
[
i
]
∗
X
[
i
]
之
前
的
个
数
−
X
[
i
]
之
前
的
数
坐
标
和
+
X
[
i
]
之
后
的
数
坐
标
和
−
X
[
i
]
∗
X
[
i
]
之
后
的
数
的
个
数
X[i] * X[i]之前的个数 - X[i]之前的数坐标和 + X[i]之后的数坐标和 -X[i]*X[i]之后的数的个数
X[i]∗X[i]之前的个数−X[i]之前的数坐标和+X[i]之后的数坐标和−X[i]∗X[i]之后的数的个数
对于X[i]之前的个数以及坐标和,只需要用两个树状数组来维护即可。
AC代码:
#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include <bitset>
#include<vector>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define endl '\n'
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> PII;
const int maxn = 2e5+200;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll f = 1; ll x = 0;char ch = getchar();while(ch>'9'||ch<'0') {if(ch=='-') f=-1; ch = getchar();}while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0', ch = getchar();return x*f; }
int dir[4][2] = { {1,0}, {-1,0},{0,1},{0,-1} };
ll sum1[maxn][2];
ll sum2[maxn][2];
ll n;
typedef struct Pos
{
ll id;
ll val;
bool operator < (const Pos &a) const
{
return val < a.val;
}
}P;
P a[maxn];
ll up = 2e4+500;
void add(ll pos, ll y, int flag) //在pos位置+y,对d
{
for(ll i=pos;i<=up;i+=lowbit(i))
sum1[i][flag] += y, sum2[i][flag] += pos*y; //从这个位置开始,包含pos项的sum都改变。 sum1是d[i]的前缀和,sum2是d[i]*i的前缀和,sum1*pos-sum2就是a的前缀和
}
void add_range(ll l, ll r,ll x, int flag)
{
add(l,x, flag), add(r+1,-x, flag);
}
ll ask_for_one(ll p, int flag)
{
ll ans = 0;
for(ll i=p;i>=1;i-=lowbit(i))
ans += sum1[i][flag];
return ans;
}
ll ask(ll p, int flag)
{
ll ans = 0;
for(ll i=p;i>=1;i-=lowbit(i))
ans += ((p+1)*sum1[i][flag] - sum2[i][flag]);
return ans;
}
ll ans[maxn];
int main()
{
n = read();
rep(i,1,n) a[i].val = read(), a[i].id = read();
sort(a+1,a+1+n);
ll sum = 0;
rep(i,1,n)
{
ll lowNum = ask(a[i].id-1,0);
ll lowSum = ask(a[i].id-1,1);
ll maNum = ask(up,0) - ask(a[i].id,0);
ll maSum = ask(up,1) - ask(a[i].id,1);
sum += (lowNum*a[i].id - lowSum)*a[i].val;
sum += (maSum - maNum*a[i].id)*a[i].val;
add_range(a[i].id,a[i].id,1,0);
add_range(a[i].id,a[i].id,a[i].id,1);
}
cout<<sum<<endl;
return 0;
}