Codeforces Round 624 (Div. 3)
F. Moving Points
二维偏序讲解,这题同时是里面的例题。
F. Moving Points
题意:
坐标轴 O X OX OX 上有 n n n 个点。第 i i i 个点位于整数点 x i x_i xi ,速度为 v i v_i vi 。可以保证没有两个点占据相同的坐标。所有 n n n 点都以恒定的速度运动, 第 i i i 点在 t t t 时刻的坐标为 t t t 可以是非整数)的坐标计算为 x i + t ⋅ v i x_i + t \cdot v_i xi+t⋅vi 。
考虑两个点 i i i 和 j j j 。让 d ( i , j ) d(i, j) d(i,j) 成为这两点在任何可能的时间时刻(甚至非整数)的最小可能距离。这意味着,如果两点 i i i 和 j j j 在某个时刻重合, d ( i , j ) d(i, j) d(i,j) 的值将是 0 0 0 。
您的任务是计算值 ∑ 1 ≤ i < j ≤ n \sum\limits_{1 \le i < j \le n} 1≤i<j≤n∑ d ( i , j ) d(i, j) d(i,j) 。 d ( i , j ) d(i, j) d(i,j) (所有点对的最小距离之和)。
思路:
讲解博客里讲的很明白了,不赘述。
具体实现是:先按照 x x x 升序排列,枚举每个点的信息。点在树状数组中的位置是速度 v i v_i vi 的相对大小值,需要离散化。树状数组只用一个也可以,不过一个节点需要记录两个信息:前面所有点到原点的距离之和 以及 前面有几个点。
code:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
int n;
pair<ll,ll> a[maxn];
ll b[maxn],cnt;
struct tree_array{
struct node{
ll dis,num;
node(ll d=0,ll m=0):dis(d),num(m){};
node operator+(node x){
return node(dis+x.dis,num+x.num);
}
}t[maxn];
int lowbit(int x){return x&-x;}
void add(int x,ll d,ll m){
for(int i=x;i<=n;i+=lowbit(i))
t[i]=t[i]+node(d,m);
}
node query(int x){
node tmp;
for(int i=x;i;i-=lowbit(i))
tmp=tmp+t[i];
return tmp;
}
pair<ll,ll> q(int x){
node tmp=query(x);
return make_pair(tmp.dis,tmp.num);
}
}tr;
int main(){
cin>>n;
for(int i=1;i<=n;i++)cin>>a[i].first;
for(int i=1;i<=n;i++)cin>>a[i].second,b[++cnt]=a[i].second;
sort(a+1,a+n+1);
sort(b+1,b+cnt+1);
cnt=unique(b+1,b+cnt+1)-b-1;
ll ans=0;
for(int i=1,idx;i<=n;i++){
idx=lower_bound(b+1,b+cnt+1,a[i].second)-b;
pair<ll,ll> tmp=tr.q(idx);
ll dis=tmp.first,num=tmp.second;
ans+=a[i].first*num-dis;
tr.add(idx,a[i].first,1);
}
cout<<ans<<endl;
return 0;
}