题目描述
约翰的N 头奶牛每年都会参加“哞哞大会”。哞哞大会是奶牛界的盛事。集会上的活动很多,比如堆干草,跨栅栏,摸牛仔的屁股等等。它们参加活动时会聚在一起,第i 头奶牛的坐标为Xi,没有两头奶牛的坐标是相同的。奶牛们的叫声很大,第i 头和第j 头奶牛交流,会发出max{Vi; Vj}×|Xi − Xj | 的音量,其中Vi 和Vj 分别是第i 头和第j 头奶牛的听力。
假设每对奶牛之间同时都在说话,请计算所有奶牛产生的音量之和是多少。
输入
4
3 1
2 5
2 6
4 3
输出
57
结论,如果一个树状数组解决不了题目,那咱们就开俩个~~!(狗头,仅限于这道题)
首先分析一下式子max{Vi; Vj}×|Xi − Xj |
乘号左边取决于两头牛听力较大者,所以我们每次计算的时候,就只计算听力比当前牛牛要小的牛牛所产生的权值部分,我们将牛牛根据听力的值,从小到大进行排序,这样就可以避免漏算和重复算。
然后分析式子的左边。绝对值当然就可以拆成两种情况。Xi<Xj和Xi>Xj,所以一头牛牛对答案的贡献就可以视为如下
V*(num1* X-Xmin+Xmax-num2* X)
num1表示横坐标小于X的牛牛数量
num2表示横坐标大于X的牛牛数量
Xmin表示横坐标小于X的牛牛的距离之和
Xmax表示横坐标大于X的牛牛的距离之和
其中num2又可以用已经出现的牛牛数量减去num1来得到
所以我们开两个树状数组来维护,num和Xmin、Xmax
#include<cstdio>
#include<iostream>
#include<cmath>
#include<vector>
#include<stack>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long ll;
const int inf=1000000000;
const ll mod=100003;
struct node {
ll val,p;
};
struct node a[20004];
ll tree1[20005],tree2[20004];//两个哦
int lowbit(int x){
return x&(-x);
}
void add1(int now,ll x){
while(now<=20000){
tree1[now]+=x;
now+=lowbit(now);
}
}
void add2(int now,ll x){
while(now<=20000){
tree2[now]+=x;
now+=lowbit(now);
}
}
ll ask1(int now){
ll ans=0;
while(now>0){
ans=ans+tree1[now];
now-=lowbit(now);
}
return ans;
}
ll ask2(int now){
ll ans=0;
while(now>0){
ans=ans+tree2[now];
now-=lowbit(now);
}
return ans;
}
bool cmp(struct node x,struct node y){
return x.val<y.val;
}
int main(){
#ifdef LOCAL
freopen(".\\a.in", "r", stdin);
#endif
int n;
memset(tree1,0,sizeof(tree1));
memset(tree2,0,sizeof(tree2));
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%lld%lld",&a[i].val,&a[i].p);
}
sort(a+1,a+n+1,cmp);//排序
ll ans=0;
for(int i=1;i<=n;i++){
ll tmp1=ask1(a[i].p);//Xmin
ll tmp2=ask1(20000)-ask1(a[i].p);//Xmax
ll num=ask2(a[i].p);//num1
ans=ans+a[i].val*(num*a[i].p-tmp1+tmp2-(i-1-num)*a[i].p);//i-1-num就是上面的num2
add1(a[i].p,a[i].p);//维护tree1
add2(a[i].p,1); //维护tree2
}
printf("%lld\n",ans);
return 0;
}